Added all SMAC Maps to dataset download function.

instadeepai · Dec 18, 2023 · f8e54a3 · f8e54a3
1 parent 8024c4d
commit f8e54a3
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 8 deletions.
diff --git a/examples/jax/jax_main.py → examples/jax/main.py b/examples/jax/jax_main.py → examples/jax/main.py
diff --git a/examples/tf2/tf2_main.py → examples/tf2/main.py b/examples/tf2/tf2_main.py → examples/tf2/main.py
diff --git a/install_environments/smacv2.sh b/install_environments/smacv2.sh
@@ -4,19 +4,19 @@
 
 export SC2PATH=~/StarCraftII
 
-echo 'StarCraftII is not installed. Installing now ...';
-wget --progress=dot:mega http://blzdistsc2-a.akamaihd.net/Linux/SC2.4.10.zip
-unzip -oP iagreetotheeula SC2.4.10.zip
-mv StarCraftII $SC2PATH
-rm -rf SC2.4.10.zip
+# echo 'StarCraftII is not installed. Installing now ...';
+# wget --progress=dot:mega http://blzdistsc2-a.akamaihd.net/Linux/SC2.4.10.zip
+# unzip -oP iagreetotheeula SC2.4.10.zip
+# mv StarCraftII $SC2PATH
+# rm -rf SC2.4.10.zip
 
 echo 'Adding SMAC maps.'
 MAP_DIR="$SC2PATH/Maps/"
 echo 'MAP_DIR is set to '$MAP_DIR
 mkdir -p $MAP_DIR
 
-wget https://github.com/oxwhirl/smac/releases/download/v0.1-beta1/SMAC_Maps.zip
-unzip SMAC_Maps.zip
+wget https://github.com/oxwhirl/smacv2/releases/download/maps/SMAC_Maps.zip
+unzip SMAC_Maps.zip -d SMAC_Maps
 mv SMAC_Maps $MAP_DIR
 rm -rf SMAC_Maps.zip 
 

diff --git a/og_marl/offline_dataset.py b/og_marl/offline_dataset.py
@@ -121,10 +121,20 @@ def __getattr__(self, name):
     "smac_v1": {
         "3m": "https://tinyurl.com/3m-dataset",
         "8m": "https://tinyurl.com/8m-dataset",
+        "5m_vs_6m": "https://tinyurl.com/5m-vs-6m-dataset",
+        "2s3z": "https://tinyurl.com/2s3z-dataset",
+        "3s5z_vs_3s6z": "ttps://tinyurl.com/3s5z-vs-3s6z-dataset3",
+        "2c_vs_64zg": "https://tinyurl.com/2c-vs-64zg-dataset",
+        "27m_vs_30m": "https://tinyurl.com/27m-vs-30m-dataset"
     },
     "smac_v2": {
         "terran_5_vs_5": "https://tinyurl.com/terran-5-vs-5-dataset",
-        "zerg_5_": "https://tinyurl.com/zerg-5-vs-5-dataset",
+        "zerg_5_vs_5": "https://tinyurl.com/zerg-5-vs-5-dataset",
+        "terran_10_vs_10": "https://tinyurl.com/terran-10-vs-10-dataset"
+    },
+    "flatland": {
+        "3_trains": "https://tinyurl.com/3trains-dataset",
+        "5_trains": "https://tinyurl.com/5trains-dataset"
     }
 }
 def download_and_unzip_dataset(env_name, scenario_name, dataset_base_dir="./datasets"):