Ruff fixes

src/gnn_tracking/utils/lightning.py

@@ -132,10 +132,10 @@ def get_model(
         freeze: Whether to freeze the model
     """
     if not chkpt_path:
-        return
+        return None
     lm = get_lightning_module(class_path, chkpt_path, freeze=freeze)
     if lm is None:
-        return
+        return None
     return lm.model
 
 

src/gnn_tracking/utils/timing.py

@@ -29,4 +29,4 @@ def timing(name="Codeblock", logger=None):
     try:
         yield
     finally:
-        logger.info(f"{name} took {t():.2f} seconds")
+        logger.info("%s took %f seconds", name, t())

tests/test_tcn_training.py

@@ -83,7 +83,7 @@ def __post_init__(self):
 
 
 @pytest.mark.parametrize("t", _test_train_test_cases)
-def test_train(tmp_path, built_graphs, t: TestTrainCase) -> None:
+def test_train(built_graphs, t: TestTrainCase) -> None:
     fix_seeds()
     _, graph_builder = built_graphs
     g = graph_builder.data_list[0]