Fix shape of angles

metatensor · Jan 31, 2025 · 593918b · 593918b
1 parent 2ad555d
commit 593918b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/featomic/tests/clebsch_gordan/rotations.py b/python/featomic/tests/clebsch_gordan/rotations.py
@@ -98,9 +98,9 @@ def __init__(self, max_angular: int, angles: Sequence[float] = None):
         self.max_angular = max_angular
         # Randomly generate Euler angles between if none are provided
         if angles is None:
-            psi = 2 * np.pi * np.random.uniform(size=1)  # psi is between 0 and 2pi
-            theta = np.pi * np.random.uniform(size=1)  # theta is between 0 and pi
-            phi = 2 * np.pi * np.random.uniform(size=1)  # phi is between 0 and 2pi
+            psi = 2 * np.pi * np.random.uniform()  # psi is between 0 and 2pi
+            theta = np.pi * np.random.uniform()  # theta is between 0 and pi
+            phi = 2 * np.pi * np.random.uniform()  # phi is between 0 and 2pi
             angles = np.array([psi, theta, phi])
         self.angles = angles
         self.rotation = cartesian_rotation(angles)