Skip to content

Commit 0dfe01a

Browse files
committed
update seeding
1 parent 7033a1b commit 0dfe01a

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

gym_maze/envs/maze.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,9 @@ def __init__(self,
2121
live_display=False,
2222
render_trace=False):
2323
"""Initialize the maze. DType: list"""
24+
# Random seed with internal gym seeding
25+
self.seed()
26+
2427
# Maze: 0: free space, 1: wall
2528
self.maze_generator = maze_generator
2629
self.maze = np.array(self.maze_generator.get_maze())
@@ -98,6 +101,11 @@ def step(self, action):
98101

99102
return self._get_obs(), reward, done, info
100103

104+
def seed(self, seed=None):
105+
self.np_random, seed = seeding.np_random(seed)
106+
107+
return [seed]
108+
101109
def reset(self):
102110
# Reset maze
103111
self.maze = np.array(self.maze_generator.get_maze())

0 commit comments

Comments
 (0)