diff --git a/reflex/utils/compat.py b/reflex/utils/compat.py index 7d2a03a11..ef5fcd3e1 100644 --- a/reflex/utils/compat.py +++ b/reflex/utils/compat.py @@ -32,6 +32,13 @@ def pydantic_v1_patch(): Yields: None when the Pydantic module is patched. """ + import pydantic + + if pydantic.__version__.startswith("1."): + # pydantic v1 is already installed + yield + return + patched_modules = [ "pydantic", "pydantic.fields", @@ -42,11 +49,6 @@ def pydantic_v1_patch(): try: import pydantic.v1 # type: ignore - if pydantic.__version__.startswith("1."): - # pydantic v1 is already installed - yield - return - sys.modules["pydantic.fields"] = pydantic.v1.fields # type: ignore sys.modules["pydantic.main"] = pydantic.v1.main # type: ignore sys.modules["pydantic.errors"] = pydantic.v1.errors # type: ignore