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