Merge pull request #989 from StanfordVL/fix-transform-test

Fix transform utils test
StanfordVL · Oct 17, 2024 · 542c590 · 542c590
2 parents bdcf4bd + a52fa12
commit 542c590
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_transform_utils.py b/tests/test_transform_utils.py
@@ -358,7 +358,7 @@ def test_quat_slerp(self):
             key_rots = R.from_quat(np.stack([q1.cpu().numpy(), q2.cpu().numpy()]))
             key_times = [0, 1]
             slerp = Slerp(key_times, key_rots)
-            scipy_q_slerp = slerp([t]).as_quat()[0].astype(NumpyTypes.FLOAT32)
+            scipy_q_slerp = slerp([t.item()]).as_quat()[0].astype(NumpyTypes.FLOAT32)
 
             assert quaternions_close(q_slerp, th.from_numpy(scipy_q_slerp))
             assert_close(th.norm(q_slerp), th.tensor(1.0))