using filled() method of the numpy masked array because using a maske…

…d array in newer jax (0.4.18) throws an error
btalamini · Oct 11, 2023 · dfd88d7 · dfd88d7
1 parent 742cd0d
commit dfd88d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimism/ReadExodusMesh.py b/optimism/ReadExodusMesh.py
@@ -37,7 +37,7 @@ def _read_coordinates(exodusDataset):
     coordsX = exodusDataset.variables['coordx'][:]
     coordsY = exodusDataset.variables['coordy'][:]
 
-    return np.column_stack([coordsX, coordsY])
+    return np.column_stack([coordsX.filled(), coordsY.filled()])
 
 
 def _read_block_conns(exodusDataset, blockOrdinal):