Small hydra config fix.

og_marl/tf2_systems/offline/continuous_bc.py

@@ -196,7 +196,7 @@ def run_experiment(cfg: DictConfig) -> None:
 
     download_and_unzip_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"])
 
-    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], cfg["task"]["dataset"])
+    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], str(cfg["task"]["dataset"]))
 
     wandb_config = {
         "system": cfg["system_name"],

og_marl/tf2_systems/offline/discrete_bc.py

@@ -215,7 +215,7 @@ def run_experiment(cfg: DictConfig) -> None:
 
     download_and_unzip_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"])
 
-    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], cfg["task"]["dataset"])
+    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], str(cfg["task"]["dataset"]))
 
     wandb_config = {
         "system": cfg["system_name"],

og_marl/tf2_systems/offline/iddpg_bc.py

@@ -305,7 +305,7 @@ def run_experiment(cfg: DictConfig) -> None:
 
     download_and_unzip_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"])
 
-    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], cfg["task"]["dataset"])
+    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], str(cfg["task"]["dataset"]))
 
     wandb_config = {
         "system": cfg["system_name"],

og_marl/tf2_systems/offline/iddpg_cql.py

@@ -427,7 +427,7 @@ def run_experiment(cfg: DictConfig) -> None:
 
     download_and_unzip_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"])
 
-    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], cfg["task"]["dataset"])
+    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], str(cfg["task"]["dataset"]))
 
     wandb_config = {
         "system": cfg["system_name"],

og_marl/tf2_systems/offline/iql_bcq.py

@@ -309,7 +309,7 @@ def run_experiment(cfg: DictConfig) -> None:
 
     download_and_unzip_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"])
 
-    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], cfg["task"]["dataset"])
+    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], str(cfg["task"]["dataset"]))
 
     wandb_config = {
         "system": cfg["system_name"],

og_marl/tf2_systems/offline/iql_cql.py

@@ -273,7 +273,7 @@ def run_experiment(cfg: DictConfig) -> None:
 
     download_and_unzip_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"])
 
-    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], cfg["task"]["dataset"])
+    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], str(cfg["task"]["dataset"]))
 
     wandb_config = {
         "system": cfg["system_name"],

og_marl/tf2_systems/offline/maddpg_cql.py

@@ -421,7 +421,7 @@ def run_experiment(cfg: DictConfig) -> None:
 
     download_and_unzip_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"])
 
-    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], cfg["task"]["dataset"])
+    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], str(cfg["task"]["dataset"]))
 
     wandb_config = {
         "system": cfg["system_name"],

og_marl/tf2_systems/offline/maicq.py

@@ -331,7 +331,7 @@ def run_experiment(cfg: DictConfig) -> None:
 
     download_and_unzip_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"])
 
-    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], cfg["task"]["dataset"])
+    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], str(cfg["task"]["dataset"]))
 
     wandb_config = {
         "system": cfg["system_name"],

og_marl/tf2_systems/offline/omar.py

@@ -503,7 +503,7 @@ def run_experiment(cfg: DictConfig) -> None:
 
     download_and_unzip_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"])
 
-    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], cfg["task"]["dataset"])
+    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], str(cfg["task"]["dataset"]))
 
     wandb_config = {
         "system": cfg["system_name"],

og_marl/tf2_systems/offline/qmix_bcq.py

@@ -346,7 +346,7 @@ def run_experiment(cfg: DictConfig) -> None:
 
     download_and_unzip_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"])
 
-    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], cfg["task"]["dataset"])
+    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], str(cfg["task"]["dataset"]))
 
     wandb_config = {
         "system": cfg["system_name"],

og_marl/tf2_systems/offline/qmix_cql.py

@@ -344,7 +344,7 @@ def run_experiment(cfg: DictConfig) -> None:
 
     download_and_unzip_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"])
 
-    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], cfg["task"]["dataset"])
+    buffer.populate_from_vault(cfg["task"]["source"], cfg["task"]["env"], cfg["task"]["scenario"], str(cfg["task"]["dataset"]))
 
     wandb_config = {
         "system": cfg["system_name"],

og_marl/tf2_systems/online/configs/maddpg.yaml

@@ -12,7 +12,7 @@ replay:
   sequence_length: 10
   sample_period: 1
   max_size: 10000
-  store_to_vault: True
+  store_to_vault: False
 
 system:
   train_period: 4

requirements/baselines.txt

@@ -1,6 +1,6 @@
-tensorflow[and-cuda]==2.15.1.*
-tensorflow-probability[tf]==0.23.0
+tensorflow[and-cuda]
+tensorflow-probability[tf]
 tensorflow_io
 dm_sonnet
 wandb
-hydra-core==1.3.2
+hydra-core

requirements/datasets.txt

@@ -1,5 +1,5 @@
 numpy
-flashbax>=0.1.2
+flashbax
 gymnasium
 requests
 dm_tree