From 45dde0072ec9325ca38495a16ac2b8e78e305b28 Mon Sep 17 00:00:00 2001 From: Khaleel Al-Adhami <khaleel.aladhami@gmail.com> Date: Wed, 15 Jan 2025 10:32:15 -0800 Subject: [PATCH] handle vars with var_type being BaseComponent --- reflex/components/core/cond.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/reflex/components/core/cond.py b/reflex/components/core/cond.py index fa63ca895..5f70fef22 100644 --- a/reflex/components/core/cond.py +++ b/reflex/components/core/cond.py @@ -10,6 +10,7 @@ from reflex.components.tags import CondTag, Tag from reflex.constants import Dirs from reflex.style import LIGHT_COLOR_MODE, resolved_color_mode from reflex.utils.imports import ImportDict, ImportVar +from reflex.utils.types import safe_issubclass from reflex.vars import VarData from reflex.vars.base import LiteralVar, Var from reflex.vars.number import ternary_operation @@ -140,7 +141,9 @@ def cond(condition: Any, c1: Any, c2: Any = None) -> Component | Var: cond_var = LiteralVar.create(condition) # If the first component is a component, create a Fragment if the second component is not set. - if isinstance(c1, BaseComponent): + if isinstance(c1, BaseComponent) or ( + isinstance(c1, Var) and safe_issubclass(c1._var_type, BaseComponent) + ): c2 = c2 if c2 is not None else Fragment.create() # Check that the second argument is valid.