From 00efda3c5b29c48ba16b004294e7c3b2dd1f606b Mon Sep 17 00:00:00 2001 From: Matteo Bettini Date: Tue, 19 Sep 2023 19:37:55 +0100 Subject: [PATCH] [Fix] Small bugfixes 2 Signed-off-by: Matteo Bettini --- benchmarl/hydra_run.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/benchmarl/hydra_run.py b/benchmarl/hydra_run.py index aa4ea4c9..3da816f2 100644 --- 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