Update main.py

Signed-off-by: Mohammed Yasin <[email protected]>
ultralytics · Jan 16, 2025 · 954a48e · 954a48e
1 parent 298a5e6
commit 954a48e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/autoimport/main.py b/autoimport/main.py
@@ -3,7 +3,6 @@
 import builtins
 import importlib
 import sys
-import time
 import types
 
 
@@ -99,7 +98,9 @@ def __exit__(self, *args):
                         self._globals[parent] = sys.modules.get(parent, LazyLoader(parent))
 
 if __name__ == "__main__":
-    # Context manager tests -----------------------------
+    import time
+
+    # Context manager
     with lazy():
         t0 = time.perf_counter()
         import numpy as np