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

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

Browse Source
This commit is contained in:
Masen Furer 2024-01-18 12:54:29 -08:00 committed by GitHub
parent 1916cf2277
commit 6fcc4fd357
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
reflex/utils/prerequisites.py
View File

@ -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)