From 14223c56baee10f8725d986e74e07827a3d4dd57 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sun, 30 Jul 2023 19:32:32 +0200 Subject: [PATCH] Fix: Controllers not getting kwargs --- fancy_gym/black_box/factory/controller_factory.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/fancy_gym/black_box/factory/controller_factory.py b/fancy_gym/black_box/factory/controller_factory.py index 8b2d865..7a4bc34 100644 --- a/fancy_gym/black_box/factory/controller_factory.py +++ b/fancy_gym/black_box/factory/controller_factory.py @@ -11,11 +11,11 @@ def get_controller(controller_type: str, **kwargs): if controller_type == "motor": return PDController(**kwargs) elif controller_type == "velocity": - return VelController() + return VelController(**kwargs) elif controller_type == "position": - return PosController() + return PosController(**kwargs) elif controller_type == "metaworld": - return MetaWorldController() + return MetaWorldController(**kwargs) else: raise ValueError(f"Specified controller type {controller_type} not supported, " f"please choose one of {ALL_TYPES}.")