Skip to content

Commit 3247742

Browse files
skbly7smsong
authored andcommitted
Bugfix: L2M2019Env seeding during __init__
1 parent cd143c2 commit 3247742

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

envs/target/v_tgt_field.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -57,12 +57,15 @@ class VTgtField(object):
5757
vtgt_space = np.array([ [-10] * 2*11*11, [10] * 2*11*11 ])
5858

5959
# -----------------------------------------------------------------------------------------------------------------
60-
def __init__(self, visualize=True, version=1, dt=.01, dt_visualize=0.5):
60+
def __init__(self, visualize=True, version=1, dt=.01, dt_visualize=0.5, seed=None):
6161
self.dt = dt
6262
self.visualize = visualize
6363
self.dt_visualize = dt_visualize
6464
self.di_visualize = int(dt_visualize/dt)
6565

66+
if seed:
67+
np.random.seed(seed)
68+
6669
# -----------------------------------------------------------------------------------------------------------------
6770
def reset(self, version=1, seed=None, pose_agent=np.array([0, 0, 0])):
6871
self.t = 0

osim/env/osim.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -459,7 +459,7 @@ def set_difficulty(self, difficulty):
459459
print("difficulty 3 for Round 2")
460460
self.spec.timestep_limit = self.time_limit
461461

462-
def __init__(self, visualize=True, integrator_accuracy=5e-5, difficulty=3, seed=0, report=None):
462+
def __init__(self, visualize=True, integrator_accuracy=5e-5, difficulty=3, seed=None, report=None):
463463
if difficulty not in [0, 1, 2, 3]:
464464
raise ValueError("difficulty level should be in [0, 1, 2, 3].")
465465
self.model_paths = {}
@@ -491,7 +491,7 @@ def __init__(self, visualize=True, integrator_accuracy=5e-5, difficulty=3, seed=
491491

492492
# create target velocity field
493493
from envs.target import VTgtField
494-
self.vtgt = VTgtField(visualize=visualize, version=self.difficulty, dt=self.osim_model.stepsize)
494+
self.vtgt = VTgtField(visualize=visualize, version=self.difficulty, dt=self.osim_model.stepsize, seed=seed)
495495
self.obs_vtgt_space = self.vtgt.vtgt_space
496496

497497
def reset(self, project=True, seed=None, init_pose=None, obs_as_dict=True):

0 commit comments

Comments
 (0)