diff --git a/main.py b/main.py
index 1abb9a5..4135c91 100644
--- a/main.py
+++ b/main.py
@@ -7,6 +7,8 @@
 
 from importlib.metadata import version
 
+import torch
+
 from lib.prune import prune_wanda_sp, prune_flap, prune_magnitude_sp, check_sparsity
 from lib.eval import eval_ppl
 
diff --git a/test/1.py b/test/1.py
new file mode 100644
index 0000000..e69de29