fix preproc test

pytorch · Sep 15, 2024 · cde51e9 · cde51e9
1 parent 68a76bb
commit cde51e9
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/test/test_libs.py b/test/test_libs.py
@@ -2857,12 +2857,8 @@ def test_load(self, selected_dataset, split):
                 break
 
     def test_minari_preproc(self, tmpdir):
-        global _MINARI_DATASETS
-        if not _MINARI_DATASETS:
-            _minari_selected_datasets()
-        selected_dataset = _MINARI_DATASETS[0]
         dataset = MinariExperienceReplay(
-            selected_dataset,
+            "D4RL/pointmaze/large-v2",
             batch_size=32,
             split_trajs=False,
             download="force",