Skip to content
Snippets Groups Projects
Commit 5e005132 authored by Tessaris Sergio's avatar Tessaris Sergio
Browse files

feat: handling missing gym module

parent fd779ece
No related branches found
No related tags found
No related merge requests found
...@@ -3,12 +3,11 @@ ...@@ -3,12 +3,11 @@
import argparse import argparse
import json import json
import gym
from gym import envs
import gym_wumpus import gym_wumpus
from gym_wumpus.envs import WumpusEnv from gym_wumpus.envs import WumpusEnv
from gym import envs, error, make
def run_episode(env: WumpusEnv): def run_episode(env: WumpusEnv):
obs = env.reset() obs = env.reset()
...@@ -39,13 +38,17 @@ def main(): ...@@ -39,13 +38,17 @@ def main():
return return
if args.file is not None: if args.file is not None:
env = gym.make('wumpus-custom-v0', desc=json.load(args.file)) env = make('wumpus-custom-v0', desc=json.load(args.file))
else: else:
try: try:
env = gym.make(args.id) env = make(args.id)
except: except error.Error as e:
env = gym.make(default_env) print('Bad Gym environment {}, using {}. Error is {}'.format(args.id, default_env, e))
run_episode(env) env = make(default_env)
if isinstance(env, WumpusEnv):
run_episode(env)
else:
print('Environment {} is not a Wumpus world.'.format(env))
if __name__ == "__main__": if __name__ == "__main__":
......
from gym.envs.registration import register try:
from gym.envs.registration import register
__version__ = "1.0.0" except ImportError as e:
raise Exception('OpenAI gym not installed, you should install wumpus package with [gym] feature.') from e
register( register(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment