code: def wrap_atari_dqn(env):
      """
      wrap the environment in atari wrappers for DQN
  
      :param env: (Gym Environment) the environment
      :return: (Gym Environment) the wrapped environment
      """
      from stable_baselines.common.atari_wrappers import wrap_deepmind
  
      return wrap_deepmind(env, frame_stack=True, scale=False)
  
imports: from stable_baselines.deepq.policies import MlpPolicy, CnnPolicy, LnMlpPolicy, LnCnnPolicy
  from stable_baselines.deepq.build_graph import build_act, build_train
  from stable_baselines.deepq.dqn import DQN
  from stable_baselines.common.buffers import ReplayBuffer, PrioritizedReplayBuffer
label_desc: Use image node_id 1 for calling a global function with example usage: wrap_atari_dqn(env) and returns: wrap_deepmind