added smoke tests of ddpg (#734)

This commit is contained in:
pzhokhov
2018-11-26 17:57:25 -08:00
committed by GitHub
parent 97e039127f
commit f3a5abaeeb
2 changed files with 22 additions and 3 deletions

View File

@@ -181,11 +181,11 @@ def parse_cmdline_kwargs(args):
def main():
def main(args):
# configure logger, disable logging in child MPI processes (with rank > 0)
arg_parser = common_arg_parser()
args, unknown_args = arg_parser.parse_known_args()
args, unknown_args = arg_parser.parse_known_args(args)
extra_args = parse_cmdline_kwargs(unknown_args)
if MPI is None or MPI.COMM_WORLD.Get_rank() == 0:
@@ -220,5 +220,7 @@ def main():
env.close()
return model
if __name__ == '__main__':
main()
main(sys.argv)