diff --git a/rllib/contrib/maddpg/maddpg.py b/rllib/contrib/maddpg/maddpg.py index ad87212ba..9a0622e2f 100644 --- a/rllib/contrib/maddpg/maddpg.py +++ b/rllib/contrib/maddpg/maddpg.py @@ -23,6 +23,9 @@ logger.setLevel(logging.INFO) # yapf: disable # __sphinx_doc_begin__ DEFAULT_CONFIG = with_common_config({ + # === Framework to run the algorithm === + "framework": "tf", + # === Settings for each individual policy === # ID of the agent controlled by this policy "agent_id": None, diff --git a/rllib/contrib/maddpg/maddpg_policy.py b/rllib/contrib/maddpg/maddpg_policy.py index 963de76fd..2d86dffe0 100644 --- a/rllib/contrib/maddpg/maddpg_policy.py +++ b/rllib/contrib/maddpg/maddpg_policy.py @@ -74,12 +74,14 @@ class MADDPGTFPolicy(MADDPGPostprocessing, TFPolicy): "Space {} is not supported.".format(space)) obs_space_n = [ - _make_continuous_space(space) for _, (_, space, _, _) in - sorted(config["multiagent"]["policies"].items()) + _make_continuous_space(space) + for _, (_, space, _, + _) in config["multiagent"]["policies"].items() ] act_space_n = [ - _make_continuous_space(space) for _, (_, _, space, _) in - sorted(config["multiagent"]["policies"].items()) + _make_continuous_space(space) + for _, (_, _, space, + _) in config["multiagent"]["policies"].items() ] # _____ Placeholders