Skip to content

Commit

Permalink
Redo
Browse files Browse the repository at this point in the history
  • Loading branch information
TristanKalloniatis committed Mar 30, 2023
1 parent 6a0c504 commit fee60f5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion jumanji/environments/logic/minesweeper/env_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -121,7 +121,7 @@ def test_minesweeper__reset(minesweeper_env: Minesweeper) -> None:


def test_minesweeper__step(minesweeper_env: Minesweeper) -> None:
"""Validates the jitted step of the environment.""
"""Validates the jitted step of the environment."""
chex.clear_trace_counter()
step_fn = chex.assert_max_traces(minesweeper_env.step, n=2)
step_fn = jax.jit(step_fn)
Expand Down

0 comments on commit fee60f5

Please sign in to comment.