Always include numpy in Python script exports as with Numpy 2.0 scala…

…rs are often represented as np.float64(...)
glue-viz · Jul 12, 2024 · 262671d · 262671d
1 parent 5e92185
commit 262671d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/glue/viewers/common/viewer.py b/glue/viewers/common/viewer.py
@@ -457,6 +457,10 @@ def export_as_script(self, filename):
 
         imports = ['from glue.core.state import load']
 
+        # Always include Numpy because as of Numpy 2.0, scalars are sometimes
+        # represented in string form as e.g. np.float64(...)
+        imports += ['import numpy as np']
+
         imports_header, header = self._script_header()
         imports.extend(imports_header)