code: def create_env_wrapper(create_env_fn):
      def inner_create(*args, **kwargs):
          env = create_env_fn(*args, **kwargs)
          env.cum_length = 0
          env.cum_reward = 0
          return env
  
      return inner_create
  
imports: import functools
  import os
  import sys
  import time
  from absl import logging
  from dopamine.discrete_domains import atari_lib
  from dopamine.discrete_domains import iteration_statistics
  from dopamine.discrete_domains import run_experiment
  from dopamine.labs.atari_100k import normalization_utils
  from dopamine.metrics import collector_dispatcher
  from dopamine.metrics import statistics_instance
  import gin
  import jax
  import numpy
  import tensorflow
label_desc: Use image node_id 1 for calling a global function with example usage: create_env_wrapper(create_env_fn) and returns: inner_create, env