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