制作和注册

Gymnasium 允许用户通过 gymnasium.make() 函数自动加载环境,并预先包装几个重要的 wrappers。为此,环境必须事先通过 gymnasium.register() 注册。要获取已注册环境的环境规范,请使用 gymnasium.spec(),要打印整个注册表,请使用 gymnasium.pprint_registry()

gymnasium.make(id: str | EnvSpec, max_episode_steps: int | None = None, disable_env_checker: bool | None = None, **kwargs: Any) Env[source]

创建先前通过 gymnasium.register() 注册的环境或 EnvSpec

要查找所有可用的环境,请使用 gymnasium.envs.registry.keys() 获取所有有效的 ID。

参数:
  • id – 环境 ID 的字符串或 EnvSpec。可选地,如果使用字符串,则可以包含要导入的模块,例如 'module:Env-v0'。这等效于首先导入模块以注册环境,然后再制作环境。

  • max_episode_steps – episode 的最大长度,可以覆盖已注册的 EnvSpec max_episode_steps,并将该值传递给 gymnasium.wrappers.TimeLimit。使用 max_episode_steps=-1 将不会将 wrapper 应用于环境。

  • disable_env_checker – 是否添加 gymnasium.wrappers.PassiveEnvCheckerNone 将默认为 EnvSpec disable_env_checker 值,否则将使用此值。

  • kwargs – 传递给环境构造函数的其他参数。

返回:

应用了 wrappers 的环境实例。

引发:

Error – 如果 idregistry 中不存在

更新日志

v1.0.0 - 删除了 autoresetapply_api_compatibility

gymnasium.make_vec(id: str | EnvSpec, num_envs: int = 1, vectorization_mode: VectorizeMode | str | None = None, vector_kwargs: dict[str, Any] | None = None, wrappers: Sequence[Callable[[Env], Wrapper]] | None = None, **kwargs) gym.vector.VectorEnv[source]

根据给定的 ID 创建向量环境。

要查找所有可用的环境,请使用 gymnasium.pprint_registry()gymnasium.registry.keys() 获取所有有效的 ID。我们将向量环境称为向量化器,而正在被向量化的环境是基础环境或向量化环境 (vectorizor(vectorized env))。

参数:
  • id – 环境名称。可选地,可以包含要导入的模块,例如 ‘module:Env-v0’

  • num_envs – 要创建的环境数量

  • vectorization_mode – 使用的向量化方法,默认为 None,这样如果环境 ID 的 spec 具有 vector_entry_point (非 None),则首先使用此方法,否则默认为 sync 以使用 gymnasium.vector.SyncVectorEnv。有效模式为 "async""sync""vector_entry_point"。建议使用 VectorizeMode 枚举而不是字符串。

  • vector_kwargs – 传递给 vectorizor 环境构造函数的其他参数,即 SyncVectorEnv(..., **vector_kwargs)

  • wrappers – 要应用于基础环境的 wrapper 函数序列。只能在 "sync""async" 模式下使用。

  • **kwargs – 传递给基础环境构造函数的其他参数。

返回:

环境的一个实例。

引发:

Error – 如果 id 不存在,则会引发错误

gymnasium.register(id: str, entry_point: EnvCreator | str | None = None, reward_threshold: float | None = None, nondeterministic: bool = False, max_episode_steps: int | None = None, order_enforce: bool = True, disable_env_checker: bool = False, additional_wrappers: tuple[WrapperSpec, ...] = (), vector_entry_point: VectorEnvCreator | str | None = None, kwargs: dict | None = None)[source]

在 gymnasium 中注册一个环境,使用 id 以与 gymnasium.make() 一起使用,其中 entry_point 是用于创建环境的字符串或可调用对象。

id 参数对应于环境的名称,语法如下:[namespace/](env_name)[-v(version)],其中 namespace-v(version) 是可选的。

它接受任意关键字参数,这些参数会传递给 EnvSpeckwargs 参数。

参数:
  • id – 环境 ID

  • entry_point – 创建环境的入口点

  • reward_threshold – 智能体被认为已学会该环境的奖励阈值

  • nondeterministic – 环境是否为非确定性的(即使知道初始种子和所有动作,也无法达到相同的状态)

  • max_episode_steps – 截断前的最大 episode 步数。如果 None,则由 gymnasium.wrappers.TimeLimit wrapper 使用。

  • order_enforce – 是否启用顺序执行器 wrapper 以确保用户以正确的顺序运行函数。 如果 True,则 gymnasium.wrappers.OrderEnforcing 将应用于环境。

  • disable_env_checker – 是否禁用应用于环境的 gymnasium.wrappers.PassiveEnvChecker

  • additional_wrappers – 要应用于环境的额外 wrapper。

  • vector_entry_point – 创建向量化环境的入口点

  • kwargs – 任意关键字参数,在初始化时传递给环境构造函数。

更新日志

v1.0.0 - autoresetapply_api_compatibility 参数已移除

gymnasium.spec(env_id: str) EnvSpec[source]

registry 中检索环境 ID 对应的 EnvSpec

参数:

env_id – 环境 ID,预期格式为 [(namespace)/]id[-v(version)]

返回:

如果环境 spec 存在,则返回环境 spec

引发:

Error – 如果环境 ID 不存在

gymnasium.pprint_registry(print_registry: dict[str, EnvSpec] = registry, *, num_cols: int = 3, exclude_namespaces: list[str] | None = None, disable_print: bool = False) str | None[source]

漂亮地打印 registry 中的所有环境。

注意

所有参数均为仅关键字参数

参数:
  • print_registry – 要打印的环境注册表。 默认为 registry

  • num_cols – 用于显示的环境排列列数。

  • exclude_namespaces – 要从打印中排除的命名空间列表。 如果只需要 ALE 环境,则很有用。

  • disable_print – 是否返回包含所有命名空间和环境 ID 的字符串,或者将字符串打印到控制台。

核心变量

class gymnasium.envs.registration.EnvSpec(id: str, entry_point: EnvCreator | str | None = None, reward_threshold: float | None = None, nondeterministic: bool = False, max_episode_steps: int | None = None, order_enforce: bool = True, disable_env_checker: bool = False, kwargs: dict = <factory>, additional_wrappers: tuple[WrapperSpec, ...] = <factory>, vector_entry_point: VectorEnvCreator | str | None = None)[source]

用于通过 gymnasium.make() 创建环境的规范。

  • id: 用于通过 gymnasium.make() 创建环境的字符串

  • entry_point: 环境位置的字符串,(import path):(environment name) 或创建环境的函数。

  • reward_threshold: 完成环境的奖励阈值。

  • nondeterministic: 如果在相同的初始状态、随机数生成器状态和动作下,环境的观察结果无法重复。

  • max_episode_steps: 环境在截断前可以执行的最大步数

  • order_enforce: 是否强制执行函数顺序,先执行 gymnasium.Env.reset(),再执行 gymnasium.Env.step()gymnasium.Env.render()

  • disable_env_checker: 是否在 gymnasium.make() 中禁用环境检查器 wrapper,默认为 False(运行环境检查器)

  • kwargs: 初始化期间传递给环境的其他关键字参数

  • additional_wrappers: 应用于环境的额外 wrapper 元组 (WrapperSpec)

  • vector_entry_point: 用于创建向量化环境的位置

更新日志

v1.0.0 - Autoreset 属性已移除

class gymnasium.envs.registration.WrapperSpec(name: str, entry_point: str, kwargs: dict[str, Any] | None)[source]

用于记录 wrapper 配置的规范。

  • name: wrapper 的名称。

  • entry_point: 用于创建 wrapper 的位置。

  • kwargs: 传递给 wrapper 的其他关键字参数。如果 wrapper 不继承自 EzPickle,则为 None

gymnasium.envs.registration.registry

Gymnasium 的全局注册表,环境规范通过 gymnasium.register() 存储在此处,gymnasium.make() 用于从中创建环境。

gymnasium.envs.registration.current_namespace

创建或注册环境时的当前命名空间。 默认情况下为 None,但使用 namespace() 可以修改此项以自动设置环境 ID 命名空间。

其他函数

gymnasium.envs.registration.get_env_id(ns: str | None, name: str, version: int | None) str[source]

给定名称以及(可选的)版本和命名空间,获取完整的环境 ID。 是 parse_env_id() 的逆操作。

参数:
  • ns – 环境命名空间

  • name – 环境名称

  • version – 环境版本

返回:

环境 ID

gymnasium.envs.registration.parse_env_id(env_id: str) tuple[str | None, str, int | None][source]

解析环境 ID 字符串格式 - [namespace/](env-name)[-v(version)],其中命名空间和版本是可选的。

参数:

env_id – 要解析的环境 ID

返回:

包含环境命名空间、环境名称和版本号的元组

引发:

Error – 如果环境 ID 不是有效的环境正则表达式

gymnasium.envs.registration.find_highest_version(ns: str | None, name: str) int | None[source]

查找注册表中给定命名空间和名称的最高已注册版本环境。

参数:
  • ns – 环境命名空间

  • name – 环境名称 (id)

返回:

具有匹配命名空间和名称的环境的最高版本,否则返回 ``None``。

gymnasium.envs.registration.namespace(ns: str)[source]

用于修改当前命名空间的上下文管理器。

gymnasium.envs.registration.load_env_creator(name: str) EnvCreator | VectorEnvCreator[source]

加载具有 "(import path):(environment name)" 样式的名称的环境,并返回环境创建函数,通常是环境类类型。

参数:

name – 环境名称

返回:

给定环境名称的环境构造函数。