Added pursuit download url.

instadeepai · Mar 8, 2024 · 4e15468 · 4e15468
1 parent b674489
commit 4e15468
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/og_marl/offline_dataset.py b/og_marl/offline_dataset.py
@@ -67,6 +67,9 @@
         "5trains": {"url": "https://s3.kao.instadeep.io/offline-marl-dataset/vaults/3trains.zip"},
         "2trains": {"url": "https://s3.kao.instadeep.io/offline-marl-dataset/vaults/5trains.zip"},
     },
+    "pettingzoo": {
+        "pursuit": {"url": "https://s3.kao.instadeep.io/offline-marl-dataset/vaults/pursuit.zip"}
+    },
 }
 
 DATASET_INFO = {