[Fix] Small bugfixes 2

Signed-off-by: Matteo Bettini <[email protected]>
facebookresearch · Sep 19, 2023 · 00efda3 · 00efda3
1 parent 2cbfa51
commit 00efda3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/benchmarl/hydra_run.py b/benchmarl/hydra_run.py
@@ -1,9 +1,8 @@
-from experiment import ExperimentConfig
 from omegaconf import DictConfig, OmegaConf
 
 from benchmarl.algorithms.common import AlgorithmConfig
 from benchmarl.environments import Task, task_config_registry
-from benchmarl.experiment import Experiment
+from benchmarl.experiment import Experiment, ExperimentConfig
 from benchmarl.models import model_config_registry
 from benchmarl.models.common import ModelConfig, parse_model_config, SequenceModelConfig