fix bug

dunnolab · Mar 5, 2024 · c783db6 · c783db6
1 parent 9b5d6df
commit c783db6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/xminigrid/benchmarks.py b/src/xminigrid/benchmarks.py
@@ -20,8 +20,8 @@
 NAME2HFFILENAME = {
     "trivial-1m": "trivial_1m_v2",
     "small-1m": "small_1m_v2",
-    "medium-1m": "medium_1m_v1_v2",
-    "medium-3m": "medium_3m_v1_v2",
+    "medium-1m": "medium_1m_v2",
+    "medium-3m": "medium_3m_v2",
     "high-1m": "high_1m_v2",
     "high-3m": "high_3m_v2",
 }