Merge branch 'master' into remove-mkl

pytorch · Apr 1, 2024 · ffa3ec2 · ffa3ec2
2 parents 71f821e + 216b348
commit ffa3ec2
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"]