Only import PyiGenerator when needed (#3291)

Speed up reflex invocation time for `--help` by 1s
This commit is contained in:
Masen Furer 2024-05-13 15:37:34 -07:00 committed by GitHub
parent 6bbd0eee8a
commit dfdf03f029
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -20,7 +20,6 @@ from reflex import constants
from reflex.config import get_config from reflex.config import get_config
from reflex.constants import CustomComponents from reflex.constants import CustomComponents
from reflex.utils import console from reflex.utils import console
from reflex.utils.pyi_generator import PyiGenerator
config = get_config() config = get_config()
custom_components_cli = typer.Typer() custom_components_cli = typer.Typer()
@ -438,6 +437,8 @@ def _run_commands_in_subprocess(cmds: list[str]) -> bool:
def _make_pyi_files(): def _make_pyi_files():
"""Create pyi files for the custom component.""" """Create pyi files for the custom component."""
from reflex.utils.pyi_generator import PyiGenerator
package_name = _get_package_config()["project"]["name"] package_name = _get_package_config()["project"]["name"]
for dir, _, _ in os.walk(f"./{package_name}"): for dir, _, _ in os.walk(f"./{package_name}"):