Fixed test pearson metric issue with old pytorch versions (#3225)

pytorch · Apr 1, 2024 · a1a7d05 · a1a7d05
1 parent 2e1a902
commit a1a7d05
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"]