original_docstring: Creates an Env with a specified brax system.
docstring: Creates an Env with a specified brax system.
docstring_tokens: Creates an Env with a specified brax system .
code: def create(env_name: str = None,
             episode_length: int = 1000,
             action_repeat: int = 1,
             auto_reset: bool = True,
             batch_size: Optional[int] = None,
             **kwargs) -> Env:
    """Creates an Env with a specified brax system."""
    assert_exists(env_name)
    if env_name in ENVS:
      env = ENVS[env_name](**kwargs)
      if episode_length is not None:
        env = wrappers.EpisodeWrapper(env, episode_length, action_repeat)
      if batch_size:
        env = wrappers.VectorWrapper(env, batch_size)
      if auto_reset:
        env = wrappers.AutoResetWrapper(env)
      return env
    else:
      return composer.create(
          env_name,
          episode_length=episode_length,
          action_repeat=action_repeat,
          auto_reset=auto_reset,
          batch_size=batch_size,
          **kwargs)
code_tokens: def create ( env_name : str = None , episode_length : int = 1000 , action_repeat : int = 1 , auto_reset : bool = True , batch_size : Optional [ int ] = None , ** kwargs ) -> Env : assert_exists ( env_name ) if env_name in ENVS : env = ENVS [ env_name ] ( ** kwargs ) if episode_length is not None : env = wrappers . EpisodeWrapper ( env , episode_length , action_repeat ) if batch_size : env = wrappers . VectorWrapper ( env , batch_size ) if auto_reset : env = wrappers . AutoResetWrapper ( env ) return env else : return composer . create ( env_name , episode_length = episode_length , action_repeat = action_repeat , auto_reset = auto_reset , batch_size = batch_size , ** kwargs )
short_docstring: Creates an Env with a specified brax system.
short_docstring_tokens: Creates an Env with a specified brax system .
comment: """Creates an Env with a specified brax system."""