Fix config reload state imports (#7028) by harsh21234i · Pull Request #7035 · reflex-dev/reflex

harsh21234i · 2026-09-02T17:07:14Z

• ## Summary
Fixes: #7028
Fix reload_config() raising StateValueError when rxconfig.py imports a
module that defines a rx.State class.

Changes

Preserve project-local dependencies during reloads within the same
RegistrationContext.
Continue evicting dependencies for new contexts or different project
roots.
Added regression tests for state reloads, fresh contexts, and cross-
project isolation.
Added a reflex-base bugfix changelog fragment.

Testing

Focused regression tests: 3 passed
Ruff check and formatting: passed

greptile-apps · 2026-09-02T17:09:51Z

Confidence Score: 5/5

The PR appears safe to merge, with no outstanding actionable findings.

Summary

Tracks imported project modules on each RegistrationContext.
Restores registered state modules and selectively reloads their containing packages.
Adds regression coverage for state preservation, stale dependency prevention, failed reload recovery, and cross-project isolation.
Adds a reflex-base bugfix news fragment.

codspeed-hq · 2026-09-02T17:11:26Z

Merging this PR will improve performance by 3.17%

⚠️

Different runtime environments detected

Some benchmarks with significant performance changes were compared across different runtime environments,
which may affect the accuracy of the results.

Open the report in CodSpeed to investigate

⚡ 1 improved benchmark
✅ 39 untouched benchmarks
⏩ 8 skipped benchmarks¹

Performance Changes

	Benchmark	`BASE`	`HEAD`	Efficiency
⚡	`test_var_access[mutable_dict]`	47 ms	45.5 ms	+3.17%

Tip

Curious why performance improved? Comment @codspeedbot explain why performance improved on this PR, or directly use the CodSpeed MCP with your agent.

_{Comparing harsh21234i:fix/config-reload-state-imports (88f15fa) with main (5d9724e)}

8 benchmarks were skipped, so the baseline results were used instead. If they were deleted from the codebase, click here and archive them to remove them from the performance reports. ↩

cubic-dev-ai

All reported issues were addressed across 3 files

_{Reply with feedback, questions, or to request a fix.

Re-trigger cubic}

FarhanAliRaza

Tested with the appmod.py / rxconfig.py pair from #7028.

On main, get_config() followed by reload_config() in one RegistrationContext raises StateValueError. The new regression test fails there with that error and passes on this branch. ruff, pyright, and tests/units/test_config.py pass on the branch.

I also ran the same reload on a forked context, the shape AppHarness uses. That still raises the same StateValueError. I removed the _config_module_deps_root comparison and re-ran the config tests to check the root guard. They all pass without it.

See the inline comments for the requested changes.

FarhanAliRaza · 2026-09-03T18:46:24Z

-            # before probing: find_spec answers from sys.modules, so modules
-            # left behind by another project directory would fake the existence
-            # check below.
+            # Always reload rxconfig, but retain its dependencies when reloading


Too long comments, explaining what is already can be seen in code. happens in many places.
we might want to clean up these.

FarhanAliRaza · 2026-09-03T19:17:34Z

    """
    ctx = RegistrationContext.ensure_context()
-    config = _get_config()
+    config = _get_config(reload_dependencies=ctx._config is None)


ctx._config is None is the wrong signal. RegistrationContext.fork() copies base_states but resets _config to None. So a forked context evicts and re-imports the state module into a context that already holds the class, and the shadow check fires again.

Repro with the appmod.py / rxconfig.py pair from #7028:

with RegistrationContext() as ctx: c.get_config() forked = ctx.fork() tok = RegistrationContext.set(forked) c.reload_config() # StateValueError: The substate class 'appmod____my_state' has been defined multiple times.

This is the path AppHarness takes (reflex/testing.py:286: fork, then reload_config()), so the harness still crashes on such a project. Please key the decision on whether the current context already holds the states those modules registered, not on whether it has a cached config. Add the fork case to the tests.

FarhanAliRaza · 2026-09-03T19:17:37Z

-            for dep in _config_module_deps:
-                sys.modules.pop(dep, None)
-            _config_module_deps.clear()
+            if reload_dependencies or _config_module_deps_root != project_root:


The _config_module_deps_root != project_root branch is not exercised. With the comparison removed, all 118 tests in tests/units/test_config.py still pass. test_get_config_evicts_dependencies_from_another_project calls _get_config() with the default reload_dependencies=True, so it never reaches this check.

Either drop _config_module_deps_root or add a test that reloads in one context after the cwd moved to a second project. Note that in that scenario a same-named state module would still hit the shadow error, so the guard may not buy anything.

FarhanAliRaza · 2026-09-03T19:17:39Z

+        )
+
+    assert reflex_base.config._get_config(first_project).app_name == "first"
+    assert reflex_base.config._get_config(second_project).app_name == "second"


Both calls use reload_dependencies=True, so this passes on main too (after the fixture is adjusted) and does not cover the new root check. To cover it, load first_project into a context, then call reload_config() from second_project in the same context.

harsh21234i · 2026-09-05T11:11:56Z

Addressed the remaining review feedback in commit 4d8f57f.

Config reloads now restore only project-local modules that define states already registered in the active RegistrationContext. Ordinary imported config dependencies are evicted and re-imported, so edited values are not stale.
Added a same-project edited-dependency regression assertion.
Extracted the test module identifiers into named constants.
The forked-context, older-context, fresh-context, project-switch, and edited-dependency cases are covered.

Verification:

Focused reload tests: 4 passed.
Full tests/units/test_config.py: 117 passed; 2 unrelated deprecation tests could not run in my borrowed environment because granian is unavailable.
Ruff check, formatting, and git diff --check pass.

Please re-review the latest commit.

cubic-dev-ai

All reported issues were addressed across 3 files (changes from recent commits).

_{Tip: Review your code locally with the cubic CLI to iterate faster.

Re-trigger cubic}

cubic-dev-ai

1 issue found across 2 files (changes from recent commits).

Prompt for AI agents (unresolved issues)


Check if these issues are valid — if so, understand the root cause of each and fix them. If appropriate, use sub-agents to investigate and fix each issue separately.


<file name="packages/reflex-base/src/reflex_base/config.py">

<violation number="1" location="packages/reflex-base/src/reflex_base/config.py:967">
P1: When a package's `__init__.py` defines a registered `rx.State`, this reload creates a replacement class while `RegistrationContext` retains the original. Skip reloading package modules that are themselves registered state modules.</violation>
</file>

_{Tip: Review your code locally with the cubic CLI to iterate faster.

Re-trigger cubic}

cubic-dev-ai · 2026-09-05T11:35:26Z

+                        sys.modules[dep] = module
+                        _config_module_deps.add(dep)
+                for dep in sorted(package_modules, key=lambda name: name.count(".")):
+                    importlib.reload(ctx._config_module_deps[dep])


P1: When a package's __init__.py defines a registered rx.State, this reload creates a replacement class while RegistrationContext retains the original. Skip reloading package modules that are themselves registered state modules.

Prompt for AI agents

Check if this issue is valid — if so, understand the root cause and fix it. At packages/reflex-base/src/reflex_base/config.py, line 967: <comment>When a package's `__init__.py` defines a registered `rx.State`, this reload creates a replacement class while `RegistrationContext` retains the original. Skip reloading package modules that are themselves registered state modules.</comment> <file context> @@ -951,10 +951,20 @@ def _get_config( sys.modules[dep] = module _config_module_deps.add(dep) + for dep in sorted(package_modules, key=lambda name: name.count(".")): + importlib.reload(ctx._config_module_deps[dep]) # only import the module if it exists. If a module spec exists then # the module exists. </file context>

Suggested change

importlib.reload(ctx._config_module_deps[dep])

if dep not in state_modules:

importlib.reload(ctx._config_module_deps[dep])

cubic-dev-ai

All reported issues were addressed across 2 files (changes from recent commits).

_{Tip: Review your code locally with the cubic CLI to iterate faster.

Re-trigger cubic}

… into fix/config-reload-state-imports

harsh21234i · 2026-09-09T05:00:54Z

hey @FarhanAliRaza please take a look whenever you get time :)

Fix config reload state imports (reflex-dev#7028)

f01c755

harsh21234i requested a review from a team as a code owner September 2, 2026 17:07

cubic-dev-ai Bot reviewed Sep 2, 2026

View reviewed changes

Comment thread packages/reflex-base/src/reflex_base/config.py Outdated

FarhanAliRaza requested changes Sep 3, 2026

View reviewed changes

Handle forked context config reloads (reflex-dev#7028)

4e05359

greptile-apps Bot reviewed Sep 5, 2026

View reviewed changes

Comment thread packages/reflex-base/src/reflex_base/config.py

Comment thread tests/units/test_config.py

Reload non-state config dependencies

4d8f57f

cubic-dev-ai Bot reviewed Sep 5, 2026

View reviewed changes

Comment thread packages/reflex-base/src/reflex_base/config.py Outdated

Preserve state package modules on reload

10ca635

greptile-apps Bot reviewed Sep 5, 2026

View reviewed changes

Comment thread packages/reflex-base/src/reflex_base/config.py Outdated

Comment thread tests/units/test_config.py Outdated

harsh21234i added 2 commits September 5, 2026 16:54

Apply CI formatter output

a6f656e

Refresh package settings during reload

5dba682

greptile-apps Bot reviewed Sep 5, 2026

View reviewed changes

Comment thread packages/reflex-base/src/reflex_base/config.py Outdated

cubic-dev-ai Bot reviewed Sep 5, 2026

View reviewed changes

Handle package reload edge cases

f49704d

cubic-dev-ai Bot reviewed Sep 5, 2026

View reviewed changes

Comment thread packages/reflex-base/src/reflex_base/config.py Outdated

harsh21234i and others added 2 commits September 5, 2026 17:22

Track imports after failed package reload

7669cec

Merge branch 'main' into fix/config-reload-state-imports

87bb1c3

greptile-apps Bot reviewed Sep 9, 2026

View reviewed changes

Comment thread tests/units/test_config.py Outdated

harsh21234i added 2 commits September 9, 2026 09:59

Name config reload test environment key

a6d5f75

Merge remote-tracking branch 'origin/fix/config-reload-state-imports'…

88f15fa

… into fix/config-reload-state-imports

	importlib.reload(ctx._config_module_deps[dep])
	if dep not in state_modules:
	importlib.reload(ctx._config_module_deps[dep])

Uh oh!

Conversation

harsh21234i commented Sep 2, 2026 • edited by cubic-dev-ai Bot Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Changes

Testing

Uh oh!

greptile-apps Bot commented Sep 2, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Confidence Score: 5/5

Summary

Uh oh!

codspeed-hq Bot commented Sep 2, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Merging this PR will improve performance by 3.17%

Performance Changes

Footnotes

Uh oh!

cubic-dev-ai Bot left a comment • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Uh oh!

FarhanAliRaza left a comment

Choose a reason for hiding this comment

Uh oh!

FarhanAliRaza Sep 3, 2026

Choose a reason for hiding this comment

Uh oh!

FarhanAliRaza Sep 3, 2026

Choose a reason for hiding this comment

Uh oh!

FarhanAliRaza Sep 3, 2026

Choose a reason for hiding this comment

Uh oh!

FarhanAliRaza Sep 3, 2026

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

harsh21234i commented Sep 5, 2026

Uh oh!

cubic-dev-ai Bot left a comment • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

cubic-dev-ai Bot left a comment • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Choose a reason for hiding this comment

Uh oh!

cubic-dev-ai Bot Sep 5, 2026

Choose a reason for hiding this comment

Uh oh!

Uh oh!

cubic-dev-ai Bot left a comment • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

harsh21234i commented Sep 9, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

2 participants

harsh21234i commented Sep 2, 2026 •

edited by cubic-dev-ai Bot

Loading

greptile-apps Bot commented Sep 2, 2026 •

edited

Loading

codspeed-hq Bot commented Sep 2, 2026 •

edited

Loading

cubic-dev-ai Bot left a comment •

edited

Loading

cubic-dev-ai Bot left a comment •

edited

Loading

cubic-dev-ai Bot left a comment •

edited

Loading

cubic-dev-ai Bot left a comment •

edited

Loading