Fix gen_envs_display.py in docs (#177)

* remove version from file * pre-commit --------- Co-authored-by: rodrigodelazcano <[email protected]>
Farama-Foundation · Sep 16, 2023 · a1b8d5e · a1b8d5e
1 parent c813b87
commit a1b8d5e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docs/_scripts/gen_envs_display.py b/docs/_scripts/gen_envs_display.py
@@ -1,4 +1,5 @@
 import os
+import re
 
 import gymnasium as gym
 from tqdm import tqdm
@@ -31,6 +32,9 @@
             env_type = split_entrypoint[-1]
             env_name = split_entrypoint[-1]
 
+        # Remove file version from env_name
+        env_name = re.sub(r"(?:_v(?P<version>\d+))", "", env_name)
+
         if env_type not in filtered_envs_by_type:
             filtered_envs_by_type[env_type] = [env_name]
         elif env_name not in filtered_envs_by_type[env_type]: