From 777e8bc60ec9f2b161419ebc25ff280b6e234445 Mon Sep 17 00:00:00 2001 From: Khaleel Al-Adhami Date: Wed, 14 Aug 2024 16:07:35 -0700 Subject: [PATCH] use get origin more --- reflex/ivars/base.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/reflex/ivars/base.py b/reflex/ivars/base.py index a953963a2..6e8982c38 100644 --- a/reflex/ivars/base.py +++ b/reflex/ivars/base.py @@ -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