Update main.py

Signed-off-by: Mohammed Yasin <[email protected]>
ultralytics · Jan 16, 2025 · cfd3608 · cfd3608
1 parent 954a48e
commit cfd3608
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/autoimport/main.py b/autoimport/main.py
@@ -41,10 +41,11 @@ def __repr__(self):
 
 class lazy:
     """Context manager for lazy imports."""
-    _lazy_modules = {}
+
     def __init__(self):
         """Initializes a context manager for lazy module imports to optimize startup time and memory usage."""
         self._original_import = builtins.__import__
+        self._lazy_modules = {}  # Store lazy modules here
         self._globals = None
 
     def __enter__(self):