fixed jnp -> mma -> jnp route

Author: NolanBlack

examples/ansys/pyvista_pymapdl.ipynb

@@ -273,6 +273,12 @@ def calculate_next_x(
         if x_max is None:
             x_max = self.x_max
 
+        # verify indputs
+        x = self.__preprocess_jnp_array(x)
+        x_min = self.__preprocess_jnp_array(x_min)
+        x_max = self.__preprocess_jnp_array(x_max)
+        constraint_values = self.__preprocess_jnp_array(constraint_values)
+        objective_gradient = self.__preprocess_jnp_array(objective_gradient)
         self.__check_input_sizes(
             x,
             x_min,
@@ -316,6 +322,15 @@ def calculate_next_x(
 
         return xmma
 
+    def __preprocess_jnp_array(
+        self,
+        x: jax.typing.ArrayLike,
+    ) -> np.array:
+        np_x = np.array(x)
+        if len(np_x.shape) == 1:
+            np_x = np_x[:, None]
+        return np_x
+
     def __check_input_sizes(
         self,
         x: jax.typing.ArrayLike,