backwards compat for inplace matmul #178

jacanchaplais · Jun 1, 2024 · 7976a69 · 7976a69
1 parent 0bbf0d3
commit 7976a69
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/graphicle/transform.py b/graphicle/transform.py
@@ -178,7 +178,11 @@ def split_momentum(
     parent = _momentum_to_numpy(momentum)
     children = np.tile(parent, (2, 1))
     children[0, :] *= z
-    children[0, :3] @= rotation_matrix(angle, axis).T
+    # backwards compatibility, switched inplace syntax for explicit ufunc
+    # children[0, :3] @= rotation_matrix(angle, axis).T
+    np.matmul(
+        children[0, :3], rotation_matrix(angle, axis).T, out=children[0, :3]
+    )
     children[1, :] -= children[0, :]
     return gcl.MomentumArray(children)