code: def init_env(max_steps, env_class, seed=None):
      config = {
          "max_steps": max_steps,
      }
      env = env_class(config)
      env.seed(seed)
      return env
  
file_path: ray/python/ray/rllib/examples/env/tests/test_matrix_sequential_social_dilemma.py
imports: import random
  from ray.rllib.examples.env.matrix_sequential_social_dilemma import IteratedPrisonersDilemma, IteratedChicken, IteratedStagHunt, IteratedBoS
label_desc: Use image node_id 2 for calling a global function with example usage: init_env(max_steps, env_class, seed) and returns: env