add back two import

Lightning-AI · Sep 13, 2024 · 4ba4848 · 4ba4848
1 parent abe05ed
commit 4ba4848
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -11,15 +11,15 @@
 from unittest.mock import MagicMock
 from tests.conftest import RunIf
 
+from lightning.fabric.accelerators import CUDAAccelerator
 if sys.platform != "darwin":
-    from lightning.fabric.accelerators import CUDAAccelerator
     from litgpt.api import (
         LLM,
         calculate_number_of_devices,
         benchmark_dict_to_markdown_table
     )
 
-    from litgpt.scripts.download import download_from_hub
+from litgpt.scripts.download import download_from_hub