fix #3366

Former-commit-id: dc20237455c36de44f8922539d7dfadd8bedb12f

src/llmtuner/webui/engine.py

@@ -1,6 +1,5 @@
-from typing import Any, Dict, Generator
+from typing import TYPE_CHECKING, Any, Dict
 
-from ..extras.packages import is_gradio_available
 from .chatter import WebChatModel
 from .common import get_model_path, list_dataset, load_config
 from .locales import LOCALES
@@ -9,8 +8,8 @@ from .runner import Runner
 from .utils import get_time
 
 
-if is_gradio_available():
-    from gradio.components import Component  # cannot use TYPE_CHECKING here
+if TYPE_CHECKING:
+    from gradio.components import Component
 
 
 class Engine:
@@ -32,7 +31,7 @@ class Engine:
 
         return output_dict
 
-    def resume(self) -> Generator[Dict[Component, Component], None, None]:
+    def resume(self):
         user_config = load_config() if not self.demo_mode else {}
         lang = user_config.get("lang", None) or "en"
 
@@ -58,7 +57,7 @@ class Engine:
             else:
                 yield self._update_component({"eval.resume_btn": {"value": True}})
 
-    def change_lang(self, lang: str) -> Dict[Component, Component]:
+    def change_lang(self, lang: str):
         return {
             elem: elem.__class__(**LOCALES[elem_name][lang])
             for elem_name, elem in self.manager.get_elem_iter()