Fix incorrect loss name index

FR-DC · Feb 21, 2024 · 33b9108 · 33b9108
1 parent 6d4ba32
commit 33b9108
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration_tests/test_pipeline.py b/tests/integration_tests/test_pipeline.py
@@ -42,5 +42,5 @@ def test_manual_segmentation_pipeline(ds):
     trainer = pl.Trainer(fast_dev_run=True)
     trainer.fit(m, datamodule=dm)
 
-    val_loss = trainer.validate(m, datamodule=dm)[0]["val_loss"]
+    val_loss = trainer.validate(m, datamodule=dm)[0]["val/ce_loss"]
     logging.debug(f"Validation score: {val_loss:.2%}")