From bbb008c95efacf8cb1a8c0ea01589f6bc9aeba09 Mon Sep 17 00:00:00 2001
From: XianBW <36835909+XianBW@users.noreply.github.com>
Date: Wed, 23 Jul 2025 19:43:25 +0800
Subject: [PATCH] fix cli command type checker (#1114)

---
 rdagent/app/cli.py                             | 2 +-
 rdagent/app/qlib_rd_loop/factor_from_report.py | 2 +-
 rdagent/app/qlib_rd_loop/model.py              | 8 +++++++-
 rdagent/app/qlib_rd_loop/quant.py              | 8 +++++++-
 4 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/rdagent/app/cli.py b/rdagent/app/cli.py
index b1c3b144..81689ac5 100644
--- a/rdagent/app/cli.py
+++ b/rdagent/app/cli.py
@@ -34,7 +34,7 @@ from rdagent.log.mle_summary import grade_summary as grade_summary
 app = typer.Typer()
 
 
-def ui(port=19899, log_dir="", debug=False, data_science=False):
+def ui(port=19899, log_dir="", debug: bool = False, data_science: bool = False):
     """
     start web app to show the log traces.
     """
diff --git a/rdagent/app/qlib_rd_loop/factor_from_report.py b/rdagent/app/qlib_rd_loop/factor_from_report.py
index f2b35bd1..984c7519 100644
--- a/rdagent/app/qlib_rd_loop/factor_from_report.py
+++ b/rdagent/app/qlib_rd_loop/factor_from_report.py
@@ -137,7 +137,7 @@ class FactorReportLoop(FactorRDLoop, metaclass=LoopMeta):
         return exp
 
 
-def main(report_folder=None, path=None, all_duration=None, checkout=True):
+def main(report_folder=None, path=None, all_duration: str | None = None, checkout: bool = True):
     """
     Auto R&D Evolving loop for fintech factors (the factors are extracted from finance reports).
 
diff --git a/rdagent/app/qlib_rd_loop/model.py b/rdagent/app/qlib_rd_loop/model.py
index d66a5a1e..c2e4ed39 100644
--- a/rdagent/app/qlib_rd_loop/model.py
+++ b/rdagent/app/qlib_rd_loop/model.py
@@ -15,7 +15,13 @@ class ModelRDLoop(RDLoop):
     skip_loop_error = (ModelEmptyError,)
 
 
-def main(path=None, step_n=None, loop_n=None, all_duration=None, checkout=True):
+def main(
+    path=None,
+    step_n: int | None = None,
+    loop_n: int | None = None,
+    all_duration: str | None = None,
+    checkout: bool = True,
+):
     """
     Auto R&D Evolving loop for fintech models
 
diff --git a/rdagent/app/qlib_rd_loop/quant.py b/rdagent/app/qlib_rd_loop/quant.py
index fddf470d..74b5ac1e 100644
--- a/rdagent/app/qlib_rd_loop/quant.py
+++ b/rdagent/app/qlib_rd_loop/quant.py
@@ -118,7 +118,13 @@ class QuantRDLoop(RDLoop):
             self.trace.hist.append((prev_out["running"], feedback))
 
 
-def main(path=None, step_n=None, loop_n=None, all_duration=None, checkout=True):
+def main(
+    path=None,
+    step_n: int | None = None,
+    loop_n: int | None = None,
+    all_duration: str | None = None,
+    checkout: bool = True,
+):
     """
     Auto R&D Evolving loop for fintech factors.
     You can continue running session by