Update gen_envs_mds.py

Farama-Foundation · Jan 14, 2025 · cf0e857 · cf0e857
1 parent cf2ecd9
commit cf0e857
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/_scripts/gen_envs_mds.py b/docs/_scripts/gen_envs_mds.py
@@ -93,7 +93,7 @@ def get_docs_from_py(file_path):
             full_env_names = [
                 full_name
                 for full_name in all_environments.keys()
-                if env_name == full_name.split("/")[1][0: full_name.rfind("_")]
+                if env_name == full_name.split("/")[1].rsplit("_", 1)[0]
             ]
             env_name_version = full_env_names[0].split("/")[1]
             if env_type == "classic" and env_name != "rps":