dodox/fancy_gym
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix: Controllers not getting kwargs

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-07-30 19:32:32 +02:00
parent 221e3fee5a
commit 14223c56ba
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
fancy_gym/black_box/factory/controller_factory.py
View File

@ -11,11 +11,11 @@ def get_controller(controller_type: str, **kwargs):
if controller_type == "motor": if controller_type == "motor":
return PDController(**kwargs) return PDController(**kwargs)
elif controller_type == "velocity": elif controller_type == "velocity":
return VelController() return VelController(**kwargs)
elif controller_type == "position": elif controller_type == "position":
return PosController() return PosController(**kwargs)
elif controller_type == "metaworld": elif controller_type == "metaworld":
return MetaWorldController() return MetaWorldController(**kwargs)
else: else:
raise ValueError(f"Specified controller type {controller_type} not supported, " raise ValueError(f"Specified controller type {controller_type} not supported, "
f"please choose one of {ALL_TYPES}.") f"please choose one of {ALL_TYPES}.")