From 954a48e139704e68f65c92fe694903faed520569 Mon Sep 17 00:00:00 2001
From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Date: Fri, 17 Jan 2025 04:18:47 +0800
Subject: [PATCH] Update main.py

Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
---
 autoimport/main.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/autoimport/main.py b/autoimport/main.py
index 7aaf3ea..1948edc 100644
--- 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