code: def make_simulated_env_fn(**env_kwargs):
    
    def env_fn(in_graph):
      class_ = SimulatedBatchEnv if in_graph else SimulatedBatchGymEnv
      return class_(**env_kwargs)
    return env_fn
docs: Returns a function creating a simulated env, in or out of graph.
  
    Args:
      **env_kwargs: kwargs to pass to the simulated env constructor.
  
    Returns:
      Function in_graph -> env.