Merge pull request #73 from aertslab/bugfix_test_return

fix incorrect return None in Crested.test(...)

src/crested/tl/_crested.py

@@ -578,9 +578,9 @@ def test(self, return_metrics: bool = False) -> dict | None:
         # Log the evaluation results
         for metric_name, metric_value in evaluation_metrics.items():
             logger.info(f"Test {metric_name}: {metric_value:.4f}")
-            return None
         if return_metrics:
             return evaluation_metrics
+        return None
 
     def get_embeddings(
         self,

tests/test_pipeline.py

@@ -79,7 +79,9 @@ def test_peak_regression():
         "tests/data/test_pipeline/test_peak_regression/checkpoints/01.keras",
         compile=True,
     )
-    trainer.test()
+
+    test_metrics = trainer.test(return_metrics=True)
+    assert isinstance(test_metrics, dict)
     trainer.predict(adata, model_name="01")
 
     trainer.predict_regions(region_idx=["chr1:1000-1600", "chr2:2000-2600"])