diff --git a/tests/pytorch/attention/test_attention_with_cp.py b/tests/pytorch/attention/test_attention_with_cp.py
index 2c0a5d9217..3c86f99cef 100644
--- a/tests/pytorch/attention/test_attention_with_cp.py
+++ b/tests/pytorch/attention/test_attention_with_cp.py
@@ -9,7 +9,6 @@
 import subprocess
 import sys
 import threading
-import time
 import pathlib
 import logging
 import copy