Update runner.py

2023-07-21 13:35:19 +08:00 · 2023-07-21 13:35:19 +08:00 · 1d1d8538c9
parent ed0e186a13
commit 1d1d8538c9
1 changed files with 1 additions and 1 deletions
--- a/src/llmtuner/webui/runner.py
+++ b/src/llmtuner/webui/runner.py
@ -26,7 +26,7 @@ class Runner:
        self.running = False

    def initialize(
-        self, lang: str, model_name: str, dataset: list
+        self, lang: str, model_name: str, dataset: List[str]
    ) -> Tuple[str, str, LoggerHandler, LogCallback]:
        if self.running:
            return None, ALERTS["err_conflict"][lang], None, None