Forced casting to np.array of the opti output.

ami-iit · Apr 6, 2023 · 3310a86 · 3310a86
1 parent e5614fe
commit 3310a86
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/hippopt/base/opti_solver.py b/src/hippopt/base/opti_solver.py
@@ -171,7 +171,7 @@ def _generate_solution_output(
                 )
             ):
                 var = dataclasses.asdict(variables)[field.name]
-                output.__setattr__(field.name, self._opti_solution.value(var))
+                output.__setattr__(field.name, np.array(self._opti_solution.value(var)))
                 continue
 
             composite_variable = variables.__getattribute__(field.name)