From a56739fd9b7a3c5f87abb75fb06c56112c148b60 Mon Sep 17 00:00:00 2001 From: Lucas Willems Date: Wed, 5 Oct 2022 19:40:49 +0200 Subject: [PATCH] Simplify make_env --- utils/env.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/utils/env.py b/utils/env.py index a866a6a0..fa4f36d4 100644 --- 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