gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-27 23:47:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Optimized log output. (#104)

Browse Source 
Optimized log output.
This commit is contained in:
WinstonLiyt
2024-07-24 17:06:52 +08:00
committed by GitHub
parent 2d4e9c41fc
commit 2b7837b774
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/app/qlib_rd_loop/factor_from_report_sh.py
+1 -1
View File
@@ -91,7 +91,7 @@ def extract_factors_and_implement(report_file_path: str) -> tuple:
with logger.tag("load_pdf_screenshot"):
pdf_screenshot = extract_first_page_screenshot_from_pdf(report_file_path)
logger.log_object(pdf_screenshot, tag="load_pdf_screenshot")
logger.log_object(pdf_screenshot)
docs_dict = load_and_process_pdfs_by_langchain(Path(report_file_path))
rdagent/scenarios/qlib/factor_experiment_loader/pdf_loader.py
+4 -4
View File
@@ -509,21 +509,21 @@ class FactorExperimentLoaderFromPDFfiles(FactorExperimentLoader):
def load(self, file_or_folder_path: Path) -> dict:
with logger.tag("docs"):
docs_dict = load_and_process_pdfs_by_langchain(Path(file_or_folder_path))
logger.log_object(docs_dict, tag="docs dict")
logger.log_object(docs_dict)
selected_report_dict = classify_report_from_dict(report_dict=docs_dict, vote_time=1)
with logger.tag("file_to_factor_result"):
file_to_factor_result = extract_factors_from_report_dict(docs_dict, selected_report_dict)
logger.log_object(file_to_factor_result, tag="file_to_factor_result")
logger.log_object(file_to_factor_result)
with logger.tag("factor_dict"):
factor_dict = merge_file_to_factor_dict_to_factor_dict(file_to_factor_result)
logger.log_object(factor_dict, tag="factor_dict")
logger.log_object(factor_dict)
with logger.tag("filtered_factor_dict"):
factor_viability, filtered_factor_dict = check_factor_viability(factor_dict)
logger.log_object(filtered_factor_dict, tag="filtered_factor_dict")
logger.log_object(filtered_factor_dict)
# factor_dict, duplication_names_list = deduplicate_factors_by_llm(factor_dict, factor_viability)