Import bugfixes in documentation scripts (#1047)

Farama-Foundation · Aug 7, 2023 · 362fa13 · 362fa13
1 parent 83d243c
commit 362fa13
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 7 deletions.
diff --git a/docs/_scripts/generate_gifs.py b/docs/_scripts/generate_gifs.py
@@ -1,6 +1,5 @@
 import subprocess
-
-from pettingzoo.test.all_modules import all_environments
+from test.all_modules import all_environments
 
 # procs = []
 for name, module in all_environments.items():

diff --git a/docs/_scripts/graph_gen.py b/docs/_scripts/graph_gen.py
@@ -2,8 +2,7 @@
 import os
 import subprocess
 from collections import defaultdict
-
-from pettingzoo.test.all_modules import all_environments
+from test.all_modules import all_environments
 
 
 def generate_cycle_words(agents, is_classic):

diff --git a/docs/_scripts/text_capture.py b/docs/_scripts/text_capture.py
@@ -1,10 +1,9 @@
 import json
 import random
+from test.all_modules import all_environments
 
 import numpy as np
 
-from pettingzoo.test.all_modules import all_environments
-
 
 def shrink_lines(data, max_height, max_width):
     lines = data.split("\r\n")[:max_height]
@@ -17,7 +16,6 @@ def shrink_lines(data, max_height, max_width):
     if "classic/" not in name:
         continue
     env = module.env()
-    # env = gin_rummy_v0.env()
     env.reset()
 
     all_text = [env.render()]