From 3c808835a59f461ccebd8f5ea02c2d491b59b5e0 Mon Sep 17 00:00:00 2001
From: Sven Mika <sven@anyscale.io>
Date: Sun, 13 Dec 2020 16:15:54 +0100
Subject: [PATCH] [RLlib] Issue 12831: AttributeError: 'NoneType' object has no
 attribute 'id' when using custom Atari env. (#12832)

---
 rllib/env/atari_wrappers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/rllib/env/atari_wrappers.py b/rllib/env/atari_wrappers.py
index 347bc6167..48ba221b3 100644
--- a/rllib/env/atari_wrappers.py
+++ b/rllib/env/atari_wrappers.py
@@ -278,7 +278,7 @@ def wrap_deepmind(env, dim=84, framestack=True):
     """
     env = MonitorEnv(env)
     env = NoopResetEnv(env, noop_max=30)
-    if "NoFrameskip" in env.spec.id:
+    if env.spec is not None and "NoFrameskip" in env.spec.id:
         env = MaxAndSkipEnv(env, skip=4)
     env = EpisodicLifeEnv(env)
     if "FIRE" in env.unwrapped.get_action_meanings():