Jelajahi Sumber

Allow app harness tests to import State subclasses (#2408)

Masen Furer 1 tahun lalu
induk
melakukan
6fcc4fd357
1 mengubah file dengan 3 tambahan dan 1 penghapusan
  1. 3 1
      reflex/utils/prerequisites.py

+ 3 - 1
reflex/utils/prerequisites.py

@@ -163,7 +163,9 @@ def get_app(reload: bool = False) -> ModuleType:
         from reflex.state import State
         from reflex.state import State
 
 
         # Reset rx.State subclasses to avoid conflict when reloading.
         # 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.
         # Reload the app module.
         importlib.reload(app)
         importlib.reload(app)