diff --git a/requirements/pytorch/check-avail-extras.py b/requirements/pytorch/check-avail-extras.py
index cebafa587a359..a9f86282bd782 100644
--- a/requirements/pytorch/check-avail-extras.py
+++ b/requirements/pytorch/check-avail-extras.py
@@ -4,7 +4,7 @@
     import matplotlib  # noqa: F401
     import omegaconf  # noqa: F401
     import rich  # noqa: F401
+    import torch
 
-    import torch  # noqa: F401
     if torch.cuda.is_available():
         import bitsandbytes  # noqa: F401