From 6fcc4fd3576760c2978d7a094061c51d270cb211 Mon Sep 17 00:00:00 2001 From: Masen Furer Date: Thu, 18 Jan 2024 12:54:29 -0800 Subject: [PATCH] Allow app harness tests to import State subclasses (#2408) --- reflex/utils/prerequisites.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/reflex/utils/prerequisites.py b/reflex/utils/prerequisites.py index 2cad261bf..249b739cf 100644 --- a/reflex/utils/prerequisites.py +++ b/reflex/utils/prerequisites.py @@ -163,7 +163,9 @@ def get_app(reload: bool = False) -> ModuleType: from reflex.state import State # Reset rx.State subclasses to avoid conflict when reloading. - State.class_subclasses.clear() + for subclass in tuple(State.class_subclasses): + if subclass.__module__ == module: + State.class_subclasses.remove(subclass) # Reload the app module. importlib.reload(app)