Commit b83f3c9e authored by drallensmith's avatar drallensmith

Add elog/record dir selection to some python agents

parent c4ffdac8
...@@ -22,7 +22,9 @@ def main(): ...@@ -22,7 +22,9 @@ def main():
parser.add_argument('--seed', type=int, default=None, parser.add_argument('--seed', type=int, default=None,
help="Python randomization seed; uses python default if 0 or not given") help="Python randomization seed; uses python default if 0 or not given")
parser.add_argument('--no-reorient', action='store_true', parser.add_argument('--no-reorient', action='store_true',
help="Do not use rew reorient action") help="Do not use the new Reorient action")
parser.add_argument('--record-dir', type=str, default='log/',
help="Set directory to use if doing HFO --record")
args=parser.parse_args() args=parser.parse_args()
if args.seed: if args.seed:
random.seed(args.seed) random.seed(args.seed)
...@@ -32,7 +34,8 @@ def main(): ...@@ -32,7 +34,8 @@ def main():
# feature set. See feature sets in hfo.py/hfo.hpp. # feature set. See feature sets in hfo.py/hfo.hpp.
hfo_env.connectToServer(hfo.LOW_LEVEL_FEATURE_SET, hfo_env.connectToServer(hfo.LOW_LEVEL_FEATURE_SET,
'bin/teams/base/config/formations-dt', args.port, 'bin/teams/base/config/formations-dt', args.port,
'localhost', 'base_left', False) 'localhost', 'base_left', False,
record_dir=args.record_dir)
if args.seed: if args.seed:
print("Python randomization seed: {0:d}".format(args.seed)) print("Python randomization seed: {0:d}".format(args.seed))
for episode in itertools.count(): for episode in itertools.count():
......
...@@ -21,6 +21,8 @@ def main(): ...@@ -21,6 +21,8 @@ def main():
help="Server port") help="Server port")
parser.add_argument('--seed', type=int, default=None, parser.add_argument('--seed', type=int, default=None,
help="Python randomization seed; uses python default if 0 or not given") help="Python randomization seed; uses python default if 0 or not given")
parser.add_argument('--record-dir', type=str, default='log/',
help="Set directory to use if doing HFO --record")
args=parser.parse_args() args=parser.parse_args()
if args.seed: if args.seed:
random.seed(args.seed) random.seed(args.seed)
...@@ -30,7 +32,8 @@ def main(): ...@@ -30,7 +32,8 @@ def main():
# feature set. See feature sets in hfo.py/hfo.hpp. # feature set. See feature sets in hfo.py/hfo.hpp.
hfo_env.connectToServer(hfo.HIGH_LEVEL_FEATURE_SET, hfo_env.connectToServer(hfo.HIGH_LEVEL_FEATURE_SET,
'bin/teams/base/config/formations-dt', args.port, 'bin/teams/base/config/formations-dt', args.port,
'localhost', 'base_left', False) 'localhost', 'base_left', False,
record_dir=args.record_dir)
if args.seed: if args.seed:
print("Python randomization seed: {0:d}".format(args.seed)) print("Python randomization seed: {0:d}".format(args.seed))
for episode in itertools.count(): for episode in itertools.count():
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment