format

Author: thunderock

graph_ml/utils/config.py

@@ -24,4 +24,4 @@ def get_formatted_os():
         return "Linux"
     if PLATFORM == "darwin":
         return "MacOS"
-    assert False, f"Unsupported platform: {PLATFORM}"
+    assert False, f"Unsupported platform: {PLATFORM}"

tests/test_gpu_available.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 import os
 
-import pytest
 
 from graph_ml.utils import config
 
@@ -32,4 +31,4 @@ def test_gpu_available():
 
 
 def test_device_count():
-    assert config.DEVICE_COUNT >= 1
+    assert config.DEVICE_COUNT >= 1