Merge pull request #1769 from braingram/masked_array

fix masked array class path for numpy 2.0
asdf-format · Mar 11, 2024 · aedf156 · aedf156
2 parents bfd7a68 + 926591d
commit aedf156
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -1,3 +1,8 @@
+3.2.0 (unreleased)
+------------------
+
+- Fix ``numpy.ma.MaskedArray`` saving for numpy 2.x [#1769]
+
 3.1.0 (2024-02-27)
 ------------------
 

diff --git a/asdf/_core/_converters/ndarray.py b/asdf/_core/_converters/ndarray.py
@@ -10,6 +10,7 @@ class NDArrayConverter(Converter):
     ]
     types = [
         np.ndarray,  # we use the type here so the extension can find the sub-classes
+        "numpy.ma.MaskedArray",  # in numpy 2.0
         "numpy.ma.core.MaskedArray",
         "asdf.tags.core.ndarray.NDArrayType",
         "asdf.tags.core.stream.Stream",