Registering default navigation and skills environments using MiniHacks environment registration function

Author: samvelyan

minihack/navigation.py

@@ -1,8 +1,8 @@
 # Copyright (c) Facebook, Inc. and its affiliates.
 
 from minihack import MiniHack
+from minihack.envs import register
 from nle import nethack
-from gym.envs import registration
 
 
 MOVE_ACTIONS = tuple(nethack.CompassDirection)
@@ -42,7 +42,7 @@ def __init__(self, *args, des_file: str = None, **kwargs):
         super().__init__(*args, des_file=des_file, **kwargs)
 
 
-registration.register(
+register(
     id="MiniHack-Navigation-Custom-v0",
     entry_point="minihack.navigation:MiniHackNavigation",
 )

minihack/skills.py

@@ -1,7 +1,8 @@
 # Copyright (c) Facebook, Inc. and its affiliates.
+
 from minihack import MiniHack
 from minihack.base import MH_DEFAULT_OBS_KEYS
-from gym.envs import registration
+from minihack.envs import register
 
 
 class MiniHackSkill(MiniHack):
@@ -51,7 +52,7 @@ def __init__(
         super().__init__(*args, des_file=des_file, **kwargs)
 
 
-registration.register(
+register(
     id="MiniHack-Skill-Custom-v0",
     entry_point="minihack.skills:MiniHackSkill",
 )