Skip to content

Commit ad7f435

Browse files
authored
Merge pull request #25 from Max-We/mu-zero
Fix render res
2 parents 0a87e1c + 4ffafec commit ad7f435

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tetris_gymnasium/wrappers/observation.py

+6
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,11 @@ def render(self) -> "RenderFrame | list[RenderFrame] | None":
8080
the environment.
8181
"""
8282
matrix = self.observation(self.env.unwrapped._get_obs())
83+
kernel = np.ones(
84+
(self.render_scaling_factor, self.render_scaling_factor, 1),
85+
dtype=np.uint8,
86+
)
87+
matrix = np.kron(matrix, kernel)
8388

8489
if self.render_mode == "human" or self.render_mode == "rgb_array":
8590
if self.render_mode == "rgb_array":
@@ -101,6 +106,7 @@ def render(self) -> "RenderFrame | list[RenderFrame] | None":
101106
self.env.unwrapped.window_name,
102107
cv2.cvtColor(matrix, cv2.COLOR_RGB2BGR),
103108
)
109+
cv2.waitKey(1)
104110

105111
return None
106112

0 commit comments

Comments
 (0)