Merge branch 'ci/bump-pt-2.6' of https://github.com/Lightning-AI/ligh…

…tning into ci/bump-pt-2.6
Lightning-AI · Jan 8, 2025 · 2d8eb16 · 2d8eb16
2 parents 6c26f48 + daa0920
commit 2d8eb16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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