laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use get origin more

Browse Source
This commit is contained in:
Khaleel Al-Adhami 2024-08-14 16:07:35 -07:00
parent 1d45f70dfe
commit 777e8bc60e
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
reflex/ivars/base.py
View File

@ -362,13 +362,9 @@ class ImmutableVar(Var, Generic[VAR_TYPE]):
from .object import ObjectVar, ToObjectOperation
from .sequence import ArrayVar, StringVar, ToArrayOperation, ToStringOperation
fixed_type = (
var_type
if var_type is None or inspect.isclass(var_type)
else get_origin(var_type)
)
fixed_type = get_origin(var_type) or var_type
fixed_output_type = output if inspect.isclass(output) else get_origin(output)
fixed_output_type = get_origin(output) or output
if fixed_output_type is dict:
return self.to(ObjectVar, output)
@ -444,7 +440,7 @@ class ImmutableVar(Var, Generic[VAR_TYPE]):
if var_type is Any:
return self
fixed_type = var_type if inspect.isclass(var_type) else get_origin(var_type)
fixed_type = get_origin(var_type) or var_type
if fixed_type is Union:
return self