allow registering a callable

openai · Sep 18, 2017 · 5af5b79 · 5af5b79
1 parent afe0c42
commit 5af5b79
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/gym/envs/registration.py b/gym/envs/registration.py
@@ -82,8 +82,11 @@ def make(self):
         if self._entry_point is None:
             raise error.Error('Attempting to make deprecated env {}. (HINT: is there a newer registered version of this env?)'.format(self.id))
 
-        cls = load(self._entry_point)
-        env = cls(**self._kwargs)
+        elif callable(self._entry_point):
+            env = self._entry_point()
+        else:
+            cls = load(self._entry_point)
+            env = cls(**self._kwargs)
 
         # Make the enviroment aware of which spec it came from.
         env.unwrapped._spec = self