|
@@ -17,7 +17,9 @@ def redraw(img):
|
|
def reset():
|
|
def reset():
|
|
if args.seed == -1:
|
|
if args.seed == -1:
|
|
seed = None
|
|
seed = None
|
|
- obs = env.reset(seed=args.seed)
|
|
|
|
|
|
+ else:
|
|
|
|
+ seed = args.seed
|
|
|
|
+ obs = env.reset(seed=seed)
|
|
|
|
|
|
if hasattr(env, 'mission'):
|
|
if hasattr(env, 'mission'):
|
|
print('Mission: %s' % env.mission)
|
|
print('Mission: %s' % env.mission)
|