fix import

xorbitsai · Aug 3, 2023 · c318f91 · c318f91
1 parent b05cdd9
commit c318f91
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/xoscar/utils.py b/python/xoscar/utils.py
@@ -40,7 +40,6 @@
     to_binary,
     to_str,
 )
-from .nvutils import cuda_count
 
 # Please refer to https://bugs.python.org/issue41451
 try:
@@ -239,6 +238,8 @@ def lazy_import(
     rename: str | None = None,
     placeholder: bool = False,
 ):
+    from .nvutils import cuda_count
+
     rename = rename or name
     prefix_name = name.split(".", 1)[0]
     globals = globals or inspect.currentframe().f_back.f_globals  # type: ignore