gnuplotlib works with numpy2

dkogan · dkogan · commit 5e663ba53e27 · 2025-03-19T11:15:57.000-07:00
Numpy2 doesn't have np.string_ anymore, so extra code is needed to support both
numpy1 and numpy2
diff --git a/gnuplotlib.py b/gnuplotlib.py
@@ -1900,7 +1900,14 @@ def write_element(e):
 labels with spaces in them
 
                     '''
-                    if type(e) is np.string_ or type(e) is np.str_:
+
+                    # Numpy 2 broke this (no more np.string_), and this extra
+                    # code is needed to work with both numpy 2 and numpy 1
+                    try:    is_string = type(e) is np.string_
+                    except: is_string = False
+                    try:    is_bytes  = type(e) is np.bytes_
+                    except: is_bytes  = False
+                    if is_string or is_bytes or type(e) is np.str_:
                         pipe.write(b'"')
                         pipe.write(str(e).encode())
                         pipe.write(b'"')
