diff --git a/mava/advanced_usage/ff_ippo_store_experience.py b/mava/advanced_usage/ff_ippo_store_experience.py index da657e9b6..0c33f33f4 100644 --- a/mava/advanced_usage/ff_ippo_store_experience.py +++ b/mava/advanced_usage/ff_ippo_store_experience.py @@ -451,6 +451,7 @@ def learner_setup( def run_experiment(_config: DictConfig) -> None: """Runs experiment.""" + _config.logger.system_name = "ff_ippo" # Logger setup config = copy.deepcopy(_config) logger = MavaLogger(config) @@ -674,7 +675,6 @@ def hydra_entry_point(cfg: DictConfig) -> None: """Experiment entry point.""" # Allow dynamic attributes. OmegaConf.set_struct(cfg, False) - cfg.logger.system_name = "ff_ippo" # Run experiment. run_experiment(cfg) diff --git a/mava/systems/mat/anakin/mat.py b/mava/systems/mat/anakin/mat.py index c7d62ac54..2973fe567 100644 --- a/mava/systems/mat/anakin/mat.py +++ b/mava/systems/mat/anakin/mat.py @@ -437,6 +437,7 @@ def learner_setup( def run_experiment(_config: DictConfig) -> float: """Runs experiment.""" + _config.logger.system_name = "mat" config = copy.deepcopy(_config) n_devices = len(jax.devices()) @@ -586,7 +587,6 @@ def hydra_entry_point(cfg: DictConfig) -> float: """Experiment entry point.""" # Allow dynamic attributes. OmegaConf.set_struct(cfg, False) - cfg.logger.system_name = "mat" eval_performance = run_experiment(cfg) jax.block_until_ready(eval_performance) diff --git a/mava/systems/ppo/anakin/ff_ippo.py b/mava/systems/ppo/anakin/ff_ippo.py index 201bd5fc0..fd97b280d 100644 --- a/mava/systems/ppo/anakin/ff_ippo.py +++ b/mava/systems/ppo/anakin/ff_ippo.py @@ -450,6 +450,7 @@ def learner_setup( def run_experiment(_config: DictConfig) -> float: """Runs experiment.""" + _config.logger.system_name = "ff_ippo" config = copy.deepcopy(_config) n_devices = len(jax.devices()) @@ -583,7 +584,6 @@ def hydra_entry_point(cfg: DictConfig) -> float: """Experiment entry point.""" # Allow dynamic attributes. OmegaConf.set_struct(cfg, False) - cfg.logger.system_name = "ff_ippo" # Run experiment. eval_performance = run_experiment(cfg) diff --git a/mava/systems/ppo/anakin/ff_mappo.py b/mava/systems/ppo/anakin/ff_mappo.py index 680e6361a..26b3e17b6 100644 --- a/mava/systems/ppo/anakin/ff_mappo.py +++ b/mava/systems/ppo/anakin/ff_mappo.py @@ -434,6 +434,7 @@ def learner_setup( def run_experiment(_config: DictConfig) -> float: """Runs experiment.""" + _config.logger.system_name = "ff_mappo" config = copy.deepcopy(_config) n_devices = len(jax.devices()) @@ -566,7 +567,6 @@ def hydra_entry_point(cfg: DictConfig) -> float: """Experiment entry point.""" # Allow dynamic attributes. OmegaConf.set_struct(cfg, False) - cfg.logger.system_name = "ff_mappo" # Run experiment. eval_performance = run_experiment(cfg) diff --git a/mava/systems/ppo/anakin/rec_ippo.py b/mava/systems/ppo/anakin/rec_ippo.py index 182382ac6..212246080 100644 --- a/mava/systems/ppo/anakin/rec_ippo.py +++ b/mava/systems/ppo/anakin/rec_ippo.py @@ -582,6 +582,7 @@ def learner_setup( def run_experiment(_config: DictConfig) -> float: """Runs experiment.""" + _config.logger.system_name = "rec_ippo" config = copy.deepcopy(_config) n_devices = len(jax.devices()) @@ -734,7 +735,6 @@ def hydra_entry_point(cfg: DictConfig) -> float: """Experiment entry point.""" # Allow dynamic attributes. OmegaConf.set_struct(cfg, False) - cfg.logger.system_name = "rec_ippo" # Run experiment. eval_performance = run_experiment(cfg) diff --git a/mava/systems/ppo/anakin/rec_mappo.py b/mava/systems/ppo/anakin/rec_mappo.py index 671d5cbc5..b995ac167 100644 --- a/mava/systems/ppo/anakin/rec_mappo.py +++ b/mava/systems/ppo/anakin/rec_mappo.py @@ -578,6 +578,7 @@ def learner_setup( def run_experiment(_config: DictConfig) -> float: """Runs experiment.""" + _config.logger.system_name = "rec_mappo" config = copy.deepcopy(_config) n_devices = len(jax.devices()) @@ -728,7 +729,6 @@ def hydra_entry_point(cfg: DictConfig) -> float: """Experiment entry point.""" # Allow dynamic attributes. OmegaConf.set_struct(cfg, False) - cfg.logger.system_name = "rec_mappo" # Run experiment. eval_performance = run_experiment(cfg) diff --git a/mava/systems/ppo/sebulba/ff_ippo.py b/mava/systems/ppo/sebulba/ff_ippo.py index 6f34c0b1a..11bfd4b26 100644 --- a/mava/systems/ppo/sebulba/ff_ippo.py +++ b/mava/systems/ppo/sebulba/ff_ippo.py @@ -549,6 +549,7 @@ def learner_setup( def run_experiment(_config: DictConfig) -> float: """Runs experiment.""" + _config.logger.system_name = "ff_ippo_sebulba" config = copy.deepcopy(_config) local_devices = jax.local_devices() @@ -739,7 +740,6 @@ def hydra_entry_point(cfg: DictConfig) -> float: """Experiment entry point.""" # Allow dynamic attributes. OmegaConf.set_struct(cfg, False) - cfg.logger.system_name = "ff_ippo_sebulba" # Run experiment. eval_performance = run_experiment(cfg) diff --git a/mava/systems/q_learning/anakin/rec_iql.py b/mava/systems/q_learning/anakin/rec_iql.py index a5a876ccd..10ec5c3ab 100644 --- a/mava/systems/q_learning/anakin/rec_iql.py +++ b/mava/systems/q_learning/anakin/rec_iql.py @@ -491,6 +491,7 @@ def update_step( def run_experiment(cfg: DictConfig) -> float: # Add runtime variables to config + cfg.logger.system_name = "rec_iql" cfg.arch.n_devices = len(jax.devices()) cfg = check_total_timesteps(cfg) @@ -627,7 +628,6 @@ def hydra_entry_point(cfg: DictConfig) -> float: """Experiment entry point.""" # Allow dynamic attributes. OmegaConf.set_struct(cfg, False) - cfg.logger.system_name = "rec_iql" # Run experiment. eval_performance = run_experiment(cfg) diff --git a/mava/systems/q_learning/anakin/rec_qmix.py b/mava/systems/q_learning/anakin/rec_qmix.py index 7dcccf75c..c2bce9796 100644 --- a/mava/systems/q_learning/anakin/rec_qmix.py +++ b/mava/systems/q_learning/anakin/rec_qmix.py @@ -545,6 +545,7 @@ def update_step( def run_experiment(cfg: DictConfig) -> float: + cfg.logger.system_name = "rec_qmix" cfg.arch.n_devices = len(jax.devices()) cfg = check_total_timesteps(cfg) @@ -676,7 +677,6 @@ def hydra_entry_point(cfg: DictConfig) -> float: """Experiment entry point.""" # Allow dynamic attributes. OmegaConf.set_struct(cfg, False) - cfg.logger.system_name = "rec_qmix" # Run experiment. eval_performance = run_experiment(cfg) diff --git a/mava/systems/sable/anakin/ff_sable.py b/mava/systems/sable/anakin/ff_sable.py index 96baf4011..24951079e 100644 --- a/mava/systems/sable/anakin/ff_sable.py +++ b/mava/systems/sable/anakin/ff_sable.py @@ -505,6 +505,7 @@ def learner_setup( def run_experiment(_config: DictConfig) -> float: """Runs experiment.""" + _config.logger.system_name = "ff_sable" config = copy.deepcopy(_config) n_devices = len(jax.devices()) @@ -655,7 +656,6 @@ def hydra_entry_point(cfg: DictConfig) -> float: """Experiment entry point.""" # Allow dynamic attributes. OmegaConf.set_struct(cfg, False) - cfg.logger.system_name = "ff_sable" # Run experiment. eval_performance = run_experiment(cfg) diff --git a/mava/systems/sable/anakin/rec_sable.py b/mava/systems/sable/anakin/rec_sable.py index c11e227d6..34c78fc7c 100644 --- a/mava/systems/sable/anakin/rec_sable.py +++ b/mava/systems/sable/anakin/rec_sable.py @@ -535,6 +535,7 @@ def learner_setup( def run_experiment(_config: DictConfig) -> float: """Runs experiment.""" + _config.logger.system_name = "rec_sable" config = copy.deepcopy(_config) n_devices = len(jax.devices()) @@ -685,7 +686,6 @@ def hydra_entry_point(cfg: DictConfig) -> float: """Experiment entry point.""" # Allow dynamic attributes. OmegaConf.set_struct(cfg, False) - cfg.logger.system_name = "rec_sable" # Run experiment. eval_performance = run_experiment(cfg) diff --git a/mava/systems/sac/anakin/ff_hasac.py b/mava/systems/sac/anakin/ff_hasac.py index 043db91d9..fe78dcf15 100644 --- a/mava/systems/sac/anakin/ff_hasac.py +++ b/mava/systems/sac/anakin/ff_hasac.py @@ -594,6 +594,7 @@ def update_step(carry: LearnerState, _: Any) -> Tuple[LearnerState, Tuple[Metric def run_experiment(cfg: DictConfig) -> float: # Add runtime variables to config + cfg.logger.system_name = "ff_hasac" cfg.arch.n_devices = len(jax.devices()) cfg = check_total_timesteps(cfg) @@ -715,7 +716,6 @@ def hydra_entry_point(cfg: DictConfig) -> float: """Experiment entry point.""" # Allow dynamic attributes. OmegaConf.set_struct(cfg, False) - cfg.logger.system_name = "ff_hasac" # Run experiment. final_return = run_experiment(cfg) diff --git a/mava/systems/sac/anakin/ff_isac.py b/mava/systems/sac/anakin/ff_isac.py index 12416d542..222654527 100644 --- a/mava/systems/sac/anakin/ff_isac.py +++ b/mava/systems/sac/anakin/ff_isac.py @@ -493,6 +493,7 @@ def update_step(carry: LearnerState, _: Any) -> Tuple[LearnerState, Tuple[Metric def run_experiment(cfg: DictConfig) -> float: # Add runtime variables to config + cfg.logger.system_name = "ff_isac" cfg.arch.n_devices = len(jax.devices()) cfg = check_total_timesteps(cfg) @@ -614,7 +615,6 @@ def hydra_entry_point(cfg: DictConfig) -> float: """Experiment entry point.""" # Allow dynamic attributes. OmegaConf.set_struct(cfg, False) - cfg.logger.system_name = "ff_isac" # Run experiment. final_return = run_experiment(cfg) diff --git a/mava/systems/sac/anakin/ff_masac.py b/mava/systems/sac/anakin/ff_masac.py index 693364d68..4e96cdbf3 100644 --- a/mava/systems/sac/anakin/ff_masac.py +++ b/mava/systems/sac/anakin/ff_masac.py @@ -511,6 +511,7 @@ def update_step(carry: LearnerState, _: Any) -> Tuple[LearnerState, Tuple[Metric def run_experiment(cfg: DictConfig) -> float: # Add runtime variables to config + cfg.logger.system_name = "ff_masac" cfg.arch.n_devices = len(jax.devices()) cfg = check_total_timesteps(cfg) @@ -627,7 +628,6 @@ def hydra_entry_point(cfg: DictConfig) -> float: """Experiment entry point.""" # Allow dynamic attributes. OmegaConf.set_struct(cfg, False) - cfg.logger.system_name = "ff_masac" # Run experiment. final_return = run_experiment(cfg)