diff --git a/tests/test_torch_compile.py b/tests/test_torch_compile.py
index 75143e1..055e496 100644
--- a/tests/test_torch_compile.py
+++ b/tests/test_torch_compile.py
@@ -11,7 +11,6 @@
 # global modules
 import unittest
 
-import numpy as np
 import torch
 
 # Local modules