minila repro for #82

epignatelli · Jul 8, 2024 · f799389 · f799389
1 parent 11599b1
commit f799389
Showing 1 changed file with 0 additions and 10 deletions.
diff --git a/tests/test_issues.py b/tests/test_issues.py
@@ -19,24 +19,14 @@
 
 from __future__ import annotations
 
-import os
 import jax
 import jax.numpy as jnp
 
 import navix as nx
 from navix import observations
 
-import matplotlib.pyplot as plt
-
 
 def test_82():
-    os.environ["JAX_OMNISTAGING"] = "0"
-
-    def render(timestep):
-        plt.imshow(timestep.observation)
-        plt.axis(False)
-        plt.savefig("test_82.png")
-
     env = nx.make(
         "Navix-DoorKey-5x5-v0",
         max_steps=100,