fixed format

Author: Ashutosh Tiwari

.pre-commit-config.yaml

@@ -18,3 +18,4 @@ repos:
     rev: v0.0.291
     hooks:
     -   id: ruff
+        args: [ --fix ]

Makefile

@@ -30,7 +30,8 @@ format:
 	@echo "formatting..."
 	@poetry install --only lint
 	@poetry run black .
-	@poetry run pre-commit run --all-files
+	# ruff fix
+	@poetry run pre-commit run --all-files --config .pre-commit-config.yaml
 
 .PHONY: run_tests
 run_tests:

graph_ml/utils/config.py

@@ -1,16 +1,15 @@
-import os, sys
+import os
 import torch
 
 
 OS = os.name
-DEVICE_TYPE = 'cpu'
-if OS == 'posix':
+DEVICE_TYPE = "cpu"
+if OS == "posix":
     if torch.cuda.is_available():
-        DEVICE_TYPE = 'cuda'
+        DEVICE_TYPE = "cuda"
     elif torch.backends.mps.is_available():
-        DEVICE_TYPE = 'mps'
+        DEVICE_TYPE = "mps"
 
 DEVICE = torch.device(DEVICE_TYPE)
 
-GPU_AVAILABLE = DEVICE_TYPE in ['cuda', 'mps']
-
+GPU_AVAILABLE = DEVICE_TYPE in ["cuda", "mps"]

graph_ml/utils/gpu_utils.py

@@ -1,3 +0,0 @@
-import torch
-
-OS = 

tests/test_gpu_available.py

@@ -1,18 +1,19 @@
 from __future__ import annotations
 import os
 
-import pytest
 from graph_ml.utils import config
 
 OS = os.name
 
+
 def test_target_os():
     assert OS == "posix"
 
 
 def test_device_type():
     assert config.DEVICE_TYPE in ["cpu", "cuda", "mps"]
 
+
 def test_gpu_available():
     if config.DEVICE_TYPE == "cuda":
         assert config.GPU_AVAILABLE
@@ -22,4 +23,3 @@ def test_gpu_available():
         assert not config.GPU_AVAILABLE
     else:
         assert False
-