fix usage without pytorch

thuml · Nov 17, 2023 · 00ea83a · 00ea83a
1 parent 4fe8d26
commit 00ea83a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/depyf/__init__.py b/depyf/__init__.py
@@ -12,12 +12,12 @@
             "Please use the nightly version of PyTorch to enable bytecode hooks.\n"
             "PyTorch nightly can be installed by: `conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly`"
         ))
+
+    from depyf.explain.enhance_logging import install, uninstall
+    from depyf.explain.enable_debugging import prepare_debug, debug
 except ImportError:
     pass
 
-from depyf.explain.enhance_logging import install, uninstall
-from depyf.explain.enable_debugging import prepare_debug, debug
-
 import os
 
 __version__ = open(f"{os.path.dirname(__file__)}/VERSION.txt").read().strip()