fixed icp test having invalid point count

BaderTim · Jan 29, 2024 · 8c8bac3 · 8c8bac3
1 parent 64c1fa2
commit 8c8bac3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/baselines/model/tests/test_icp.py b/baselines/model/tests/test_icp.py
@@ -86,6 +86,6 @@ def rotation_matrix(axis, theta):
 def test_icp_model():
     K = 2
     model = ICP(K=K)
-    x = torch.randn(8, K, 50_200, 3) # (B, L, C, N)
+    x = torch.randn(8, K, 19_200, 3) # (B, L, C, N)
     out = model(x)
     assert out.shape == (8, 5)