Update tests/pointcloud/detection/test_data.py

Lightning-Universe · Jul 16, 2021 · b3813a4 · b3813a4
1 parent f3d0272
commit b3813a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/pointcloud/detection/test_data.py b/tests/pointcloud/detection/test_data.py
@@ -57,4 +57,4 @@ def training_step(self, batch, batch_idx: int):
     predictions = model.predict([join(predict_path, "scans/000000.bin")])
     assert torch.stack(predictions[0][DefaultDataKeys.INPUT]).shape[1] == 4
     assert len(predictions[0][DefaultDataKeys.PREDS]) == 158
-    assert predictions[0][DefaultDataKeys.PREDS][0].__dict__["identifier"] == 'box:9'
+    assert predictions[0][DefaultDataKeys.PREDS][0].__dict__["identifier"] == 'box:1'