[util] Account for scalar results

pulp-platform · Jun 19, 2024 · 0bbe21d · 0bbe21d
1 parent da74393
commit 0bbe21d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/util/sim/data_utils.py b/util/sim/data_utils.py
@@ -112,6 +112,11 @@ def flatten(array):
         return array.numpy().flatten()
     elif isinstance(array, list):
         return np.array(array).flatten()
+    # if scalar return it as a list
+    elif isinstance(array, np.generic):
+        return np.array([array]).flatten()
+    else:
+        raise TypeError(f"Unsupported type: {type(array)}")
 
 
 def _variable_attributes(alignment=None, section=None):