Commit c4ffdac8 authored by drallensmith's avatar drallensmith

Made REORIENT suppressable by an option for high_action_random_agent.py

parent b2def020
...@@ -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('--no-reorient', action='store_true',
help="Do not use rew reorient action")
args=parser.parse_args() args=parser.parse_args()
if args.seed: if args.seed:
random.seed(args.seed) random.seed(args.seed)
...@@ -49,7 +51,8 @@ def main(): ...@@ -49,7 +51,8 @@ def main():
hfo_env.act(hfo.DRIBBLE) hfo_env.act(hfo.DRIBBLE)
num_had_ball += 1 num_had_ball += 1
# 8 is frozen; rest are self or ball position/velocity valid # 8 is frozen; rest are self or ball position/velocity valid
elif (state[8] > 0) or (min(state[0],state[1],state[50],state[54]) < 0): elif (((state[8] > 0) or (min(state[0],state[1],state[50],state[54]) < 0)) and
not args.no_reorient):
hfo_env.act(hfo.REORIENT) hfo_env.act(hfo.REORIENT)
num_reorient += 1 num_reorient += 1
else: else:
......
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