minor

pculbertson · May 25, 2024 · 8c93218 · 8c93218
1 parent dab1b61
commit 8c93218
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/argus/data.py b/argus/data.py
@@ -211,7 +211,7 @@ def __getitem__(self, idx: int) -> dict:
     import cv2
     import tyro
 
-    dataset_cfg = CameraCubePoseDatasetConfig(dataset_path=ROOT + "/outputs/data/cube_unity_data_small.hdf5")
+    dataset_cfg = CameraCubePoseDatasetConfig(dataset_path=ROOT + "/outputs/data/cube_unity_data_small")
     augmentation_cfg = tyro.cli(AugmentationConfig)
     train_dataset = CameraCubePoseDataset(dataset_cfg, train=True)