wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-09-12 12:51:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

[RLlib] Tf2.x native. (#8752)

Browse Source
This commit is contained in:
Sven Mika
2020-07-11 22:06:35 +02:00
committed by GitHub
parent 5c853eaa6a
commit fcdf410ae1
45 changed files with 359 additions and 202 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rllib/tests/test_supported_multi_agent.py
+3 -2
View File
@@ -15,7 +15,8 @@ def check_support_multiagent(alg, config):
lambda _: MultiAgentCartPole({"num_agents": 2}))
config["log_level"] = "ERROR"
for fw in framework_iterator(config):
if fw == "tfe" and alg in ["A3C", "APEX", "APEX_DDPG", "IMPALA"]:
if fw in ["tf2", "tfe"] and \
alg in ["A3C", "APEX", "APEX_DDPG", "IMPALA"]:
continue
if alg in ["DDPG", "APEX_DDPG", "SAC"]:
a = get_agent_class(alg)(
@@ -123,6 +124,6 @@ if __name__ == "__main__":
import sys
# One can specify the specific TestCase class to run.
# None for all unittest.TestCase classes in this file.
class_ = sys.argv[1] if len(sys.argv) > 0 else None
class_ = sys.argv[1] if len(sys.argv) > 1 else None
sys.exit(pytest.main(
["-v", __file__ + ("" if class_ is None else "::" + class_)]))