From 216b348eaf45fb68592c805146662f0254453ec8 Mon Sep 17 00:00:00 2001 From: vfdev Date: Mon, 1 Apr 2024 11:57:33 +0200 Subject: [PATCH] Fixed test pearson metric issue with old pytorch versions (#3225) --- tests/ignite/metrics/regression/test_pearson_correlation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/ignite/metrics/regression/test_pearson_correlation.py b/tests/ignite/metrics/regression/test_pearson_correlation.py index 1a330b3a67e..b16a54c246e 100644 --- a/tests/ignite/metrics/regression/test_pearson_correlation.py +++ b/tests/ignite/metrics/regression/test_pearson_correlation.py @@ -120,8 +120,8 @@ def update_fn(engine: Engine, batch): m = PearsonCorrelation() m.attach(engine, "corr") - np_y = y.ravel().numpy() - np_y_pred = y_pred.ravel().numpy() + np_y = y.numpy().ravel() + np_y_pred = y_pred.numpy().ravel() data = list(range(y_pred.shape[0] // batch_size)) corr = engine.run(data, max_epochs=1).metrics["corr"]