浏览代码

fix appharness app_source typing (#3777)

benedikt-bartscher 9 月之前
父节点
当前提交
b58ce1082e
共有 1 个文件被更改,包括 4 次插入2 次删除
  1. 4 2
      reflex/testing.py

+ 4 - 2
reflex/testing.py

@@ -115,7 +115,7 @@ class AppHarness:
 
 
     app_name: str
     app_name: str
     app_source: Optional[
     app_source: Optional[
-        types.FunctionType | types.ModuleType | str | functools.partial
+        types.FunctionType | types.ModuleType | str | functools.partial[Any]
     ]
     ]
     app_path: pathlib.Path
     app_path: pathlib.Path
     app_module_path: pathlib.Path
     app_module_path: pathlib.Path
@@ -134,7 +134,9 @@ class AppHarness:
     def create(
     def create(
         cls,
         cls,
         root: pathlib.Path,
         root: pathlib.Path,
-        app_source: Optional[types.FunctionType | types.ModuleType | str] = None,
+        app_source: Optional[
+            types.FunctionType | types.ModuleType | str | functools.partial[Any]
+        ] = None,
         app_name: Optional[str] = None,
         app_name: Optional[str] = None,
     ) -> "AppHarness":
     ) -> "AppHarness":
         """Create an AppHarness instance at root.
         """Create an AppHarness instance at root.