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
parent 7727c14758
commit d520b34762
No known key found for this signature in database
GPG Key ID: 2AE2BD5531FF94F4

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}"):