Simplify make_env

lcswillems · Oct 5, 2022 · a56739f · a56739f
1 parent 7dcb2be
commit a56739f
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/utils/env.py b/utils/env.py
@@ -2,9 +2,6 @@
 
 
 def make_env(env_key, seed=None, render_mode=None):
-    if render_mode is None:
-        env = gym.make(env_key)
-    else:
-        env = gym.make(env_key, render_mode=render_mode)
+    env = gym.make(env_key, render_mode=render_mode)
     env.reset(seed=seed)
     return env