From 78159c51f2e3448f2d2a86c38d85c645e58df0d8 Mon Sep 17 00:00:00 2001 From: RicardoRei Date: Mon, 8 Jan 2024 12:19:08 +0000 Subject: [PATCH] removed breakpoint left behind --- tests/integration/models/test_regression_metric.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/integration/models/test_regression_metric.py b/tests/integration/models/test_regression_metric.py index 8436b48..7a7058e 100644 --- a/tests/integration/models/test_regression_metric.py +++ b/tests/integration/models/test_regression_metric.py @@ -84,6 +84,5 @@ def test_training(self): predictions = trainer.predict( ckpt_path="best", dataloaders=dataloader, return_predictions=True ) - breakpoint() y_hat = torch.cat([p["scores"] for p in predictions], dim=0).tolist() assert pearsonr(y_hat, y)[0] > 0.85