Put numpy.string_ back in for numpy<2.

esheldon · May 17, 2024 · a99e6bd · a99e6bd
1 parent 12b72eb
commit a99e6bd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/fitsio/util.py b/fitsio/util.py
@@ -41,7 +41,10 @@ def cfitsio_version(asfloat=False):
             numpy.uint64, numpy.int64)
 
 # different for py3
-_stypes += (numpy.str_, )
+if numpy.lib.NumpyVersion(numpy.__version__) >= "2.0.0":
+    _stypes += (numpy.str_,)
+else:
+    _stypes += (numpy.string_, numpy.str_,)
 
 # for header keywords
 _ftypes = (float, numpy.float32, numpy.float64)