New Structure Demo (#120)

better demo
---------

Co-authored-by: Young <afe.young@gmail.com>
Co-authored-by: Taozhi Wang <taozhi.mark.wang@gmail.com>
Co-authored-by: you-n-g <you-n-g@users.noreply.github.com>
Co-authored-by: cyncyw <47289405+taozhiwang@users.noreply.github.com>

rdagent/app/model_extraction_and_code/GeneralModel.py

@@ -1,13 +1,12 @@
 from pathlib import Path
 
-from rdagent.components.coder.model_coder.model import (
-    ModelExperiment,
-)
+from rdagent.components.coder.model_coder.model import ModelExperiment
 from rdagent.core.prompts import Prompts
 from rdagent.core.scenario import Scenario
 
 prompt_dict = Prompts(file_path=Path(__file__).parent / "prompts.yaml")
 
+
 class GeneralModelScenario(Scenario):
     @property
     def background(self) -> str:
@@ -28,10 +27,10 @@ class GeneralModelScenario(Scenario):
     @property
     def simulator(self) -> str:
         return prompt_dict["general_model_simulator"]
-    
+
     @property
-    def rich_style_description(self)->str:
-        return '''
+    def rich_style_description(self) -> str:
+        return """
 # General Model Scenario
 
 ## Overview
@@ -61,7 +60,7 @@ This scenario automates the development of PyTorch models by reading academic pa
 - **Time-Series Data:** Sequential data points indexed in time order, useful for forecasting and temporal pattern recognition.
 - **Graph Data:** Data structured as nodes and edges, suitable for network analysis and relational tasks.
 
-        '''
+        """
 
     def get_scenario_all_desc(self) -> str:
         return f"""Background of the scenario:

rdagent/app/model_extraction_and_code/model_extraction_and_implementation.py

@@ -3,18 +3,24 @@ from dotenv import load_dotenv
 
 load_dotenv(override=True)
 
+import fire
+
+from rdagent.app.model_extraction_and_code.GeneralModel import GeneralModelScenario
 from rdagent.components.coder.model_coder.task_loader import (
     ModelExperimentLoaderFromPDFfiles,
 )
-from rdagent.scenarios.qlib.developer.model_coder import QlibModelCoSTEER
-from rdagent.app.model_extraction_and_code.GeneralModel import GeneralModelScenario
-from rdagent.components.document_reader.document_reader import extract_first_page_screenshot_from_pdf
+from rdagent.components.document_reader.document_reader import (
+    extract_first_page_screenshot_from_pdf,
+)
 from rdagent.log import rdagent_logger as logger
-import fire
+from rdagent.scenarios.qlib.developer.model_coder import QlibModelCoSTEER
 
-def extract_models_and_implement(report_file_path: str = "/home/v-xisenwang/RD-Agent/rdagent/app/model_extraction_and_code/test_doc1.pdf") -> None:
+
+def extract_models_and_implement(
+    report_file_path: str = "/home/v-xisenwang/RD-Agent/rdagent/app/model_extraction_and_code/test_doc1.pdf",
+) -> None:
     with logger.tag("r"):
-    # Save Relevant Images 
+        # Save Relevant Images
         img = extract_first_page_screenshot_from_pdf(report_file_path)
         logger.log_object(img, tag="pdf_image")
         scenario = GeneralModelScenario()
@@ -26,5 +32,6 @@ def extract_models_and_implement(report_file_path: str = "/home/v-xisenwang/RD-A
         logger.log_object(exp, tag="developed_experiment")
     return exp
 
+
 if __name__ == "__main__":
     fire.Fire(extract_models_and_implement)