From 068292f749c3b3d92071fcd4db9bf12edc4691e0 Mon Sep 17 00:00:00 2001 From: WinstonLiyt <104308117+WinstonLiyt@users.noreply.github.com> Date: Wed, 24 Jul 2024 17:06:52 +0800 Subject: [PATCH] Optimized log output. (#104) Optimized log output. --- rdagent/app/qlib_rd_loop/factor_from_report_sh.py | 2 +- .../scenarios/qlib/factor_experiment_loader/pdf_loader.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/rdagent/app/qlib_rd_loop/factor_from_report_sh.py b/rdagent/app/qlib_rd_loop/factor_from_report_sh.py index 94396ede..c8cc57e4 100644 --- a/rdagent/app/qlib_rd_loop/factor_from_report_sh.py +++ b/rdagent/app/qlib_rd_loop/factor_from_report_sh.py @@ -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)) diff --git a/rdagent/scenarios/qlib/factor_experiment_loader/pdf_loader.py b/rdagent/scenarios/qlib/factor_experiment_loader/pdf_loader.py index 5a265b4f..d3d4f471 100644 --- a/rdagent/scenarios/qlib/factor_experiment_loader/pdf_loader.py +++ b/rdagent/scenarios/qlib/factor_experiment_loader/pdf_loader.py @@ -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)