From dfdf03f0296a52e62feed64d66a8c95555467908 Mon Sep 17 00:00:00 2001
From: Masen Furer <m_github@0x26.net>
Date: Mon, 13 May 2024 15:37:34 -0700
Subject: [PATCH] Only import PyiGenerator when needed (#3291)

Speed up reflex invocation time for `--help` by 1s
---
 reflex/custom_components/custom_components.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/reflex/custom_components/custom_components.py b/reflex/custom_components/custom_components.py
index 79ab26d93..ec35fb5c4 100644
--- a/reflex/custom_components/custom_components.py
+++ b/reflex/custom_components/custom_components.py
@@ -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}"):