ensure env is not glibal var in vis.py

syrkis · Jun 24, 2024 · 4eaa3c2 · 4eaa3c2
1 parent e3ae880
commit 4eaa3c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/parabellum/vis.py b/parabellum/vis.py
@@ -57,7 +57,7 @@ def animate(self, save_fname: str = "output/parabellum.mp4"):
                     state_seq, action_seq, save_fname.replace(".mp4", f"_{i}.mp4")
                 )
         else:
-            state_seq = env.expand_state_seq(self.state_seq)
+            state_seq = self.env.expand_state_seq(self.state_seq)
             self.animate_one(state_seq, self.action_seq, save_fname)
 
     def animate_one(self, state_seq, action_seq, save_fname):