From 0182fe1877e95b2ef0a82747c20bed1523fb5a3f Mon Sep 17 00:00:00 2001
From: NicoBach <43953203+NicoBach@users.noreply.github.com>
Date: Tue, 6 Aug 2019 01:03:19 +0200
Subject: [PATCH] entrypoint variable made public (#970)

---
 baselines/run.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/baselines/run.py b/baselines/run.py
index 13f7f6c..e1b11a5 100644
--- a/baselines/run.py
+++ b/baselines/run.py
@@ -32,7 +32,7 @@ except ImportError:
 _game_envs = defaultdict(set)
 for env in gym.envs.registry.all():
     # TODO: solve this with regexes
-    env_type = env._entry_point.split(':')[0].split('.')[-1]
+    env_type = env.entry_point.split(':')[0].split('.')[-1]
     _game_envs[env_type].add(env.id)
 
 # reading benchmark names directly from retro requires
@@ -126,7 +126,7 @@ def get_env_type(args):
 
     # Re-parse the gym registry, since we could have new envs since last time.
     for env in gym.envs.registry.all():
-        env_type = env._entry_point.split(':')[0].split('.')[-1]
+        env_type = env.entry_point.split(':')[0].split('.')[-1]
         _game_envs[env_type].add(env.id)  # This is a set so add is idempotent
 
     if env_id in _game_envs.keys():