Browse Source

fix vars.pyi manually

Masen Furer 1 year ago
parent
commit
3e41095df0
1 changed files with 11 additions and 2 deletions
  1. 11 2
      reflex/vars.pyi

+ 11 - 2
reflex/vars.pyi

@@ -9,7 +9,7 @@ from reflex.base import Base as Base
 from reflex.state import State as State
 from reflex.state import State as State
 from reflex.state import BaseState as BaseState
 from reflex.state import BaseState as BaseState
 from reflex.utils import console as console, format as format, types as types
 from reflex.utils import console as console, format as format, types as types
-from reflex.utils.imports import ImportVar
+from reflex.utils.imports import ImportDict, ImportList, ImportVar
 from types import FunctionType
 from types import FunctionType
 from typing import (
 from typing import (
     Any,
     Any,
@@ -35,9 +35,18 @@ def _extract_var_data(value: Iterable) -> list[VarData | None]: ...
 
 
 class VarData(Base):
 class VarData(Base):
     state: str
     state: str
-    imports: dict[str, set[ImportVar]]
+    imports: ImportList
     hooks: Dict[str, None]
     hooks: Dict[str, None]
     interpolations: List[Tuple[int, int]]
     interpolations: List[Tuple[int, int]]
+    def __init__(
+        self,
+        imports: ImportList
+        | List[ImportVar | Dict[str, Optional[Union[str, bool]]]]
+        | ImportDict
+        | Dict[str, set[ImportVar]]
+        | None = None,
+        **kwargs,
+    ): ...
     @classmethod
     @classmethod
     def merge(cls, *others: VarData | None) -> VarData | None: ...
     def merge(cls, *others: VarData | None) -> VarData | None: ...