Force ndarray 0.15 to have rayon feature enabled

gauteh · Aug 12, 2024 · 7da718f · 7da718f
1 parent b1abf36
commit 7da718f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -36,6 +36,8 @@ hdf5-sys = { package = "hdf5-metno-sys", version = "0.9.1" }
 log = "0.4"
 rayon = "1.10"
 ndarray = { version = "0.16", features = [ "rayon" ] }
+# Remove when https://github.com/PyO3/rust-numpy/pull/439 is addressed
+ndarray_0_15 = { package = "ndarray", version = "0.15", features = ["rayon"] }
 pyo3 = { version = "0.21", optional = true, features = ["anyhow", "auto-initialize", "abi3-py39"] }
 numpy = { version = "0.21.0", optional = true }
 netcdf = { version = "0.10.4", optional = true }

diff --git a/src/python/mod.rs b/src/python/mod.rs
@@ -164,7 +164,7 @@ impl Dataset {
         let arr = arr.downcast::<PyArrayDyn<T>>().unwrap();
 
         let mut v = unsafe { arr.as_array_mut() };
-        v.mapv_inplace(|v| if v == cond { fv } else { v });
+        v.par_mapv_inplace(|v| if v == cond { fv } else { v });
     }
 }