More help getting (3,) arrays into (1,3) form

MintonGroup · Dec 14, 2022 · c3583b8 · c3583b8
1 parent a87ef23
commit c3583b8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/swiftest/swiftest/simulation_class.py b/python/swiftest/swiftest/simulation_class.py
@@ -2407,6 +2407,8 @@ def input_to_array_3d(val,n=None):
                 elif n == 1:
                     if val.shape != (1,3) and val.shape != (3,):
                         raise ValueError(f"Argument is an incorrect shape. Expected {(n,3)} or {(3,1)}. Got {val.shape} instead")
+                    elif val.shape == (3,):
+                        val = np.expand_dims(val,axis=0)
                 elif val.shape != (n,3) or val.shape != (3,n):
                     raise ValueError(f"Argument is an incorrect shape. Expected {(n,3)} or {(3,n)}. Got {val.shape} instead")
                 elif val.shape == (3,n):