code: def init_env(
      max_steps, batch_size, env_class, seed=None, grid_size=3
  ):
      config = {
          "max_steps": max_steps,
          "batch_size": batch_size,
          "grid_size": grid_size,
      }
      env = env_class(config)
      env.seed(seed)
      return env
  
imports: import copy
  import random
  import numpy
  from ray.rllib.examples.env.coin_game_vectorized_env import VectorizedCoinGame, AsymVectorizedCoinGame
  from ray.rllib.examples.env.tests.test_coin_game_non_vectorized_env import assert_obs_is_symmetrical
label_desc: Use image node_id 3 for calling a global function with example usage: init_env(max_steps, batch_size, env_class, seed, grid_size) and returns: env