def METHOD_NAME(env_name: str) -> jax_types.EnvironmentFactory:
  def environment_factory(seed: int) -> dm_env.Environment:
    del seed
    return multigrid_wrapper.make_multigrid_environment(env_name)
  return environment_factory