Compute statistics for the current environment. Parameters ---------- env : ``gym.wrappers`` or ``gym.envs`` instance The environment to evaluate. Returns ------- env_info : dict A dictionary containing information about the action and observation
(env)
| 428 | |
| 429 | |
| 430 | def env_stats(env): |
| 431 | """ |
| 432 | Compute statistics for the current environment. |
| 433 | |
| 434 | Parameters |
| 435 | ---------- |
| 436 | env : ``gym.wrappers`` or ``gym.envs`` instance |
| 437 | The environment to evaluate. |
| 438 | |
| 439 | Returns |
| 440 | ------- |
| 441 | env_info : dict |
| 442 | A dictionary containing information about the action and observation |
| 443 | spaces of `env`. |
| 444 | """ |
| 445 | md_action, md_obs, tuple_action, tuple_obs = is_multidimensional(env) |
| 446 | cont_action, cont_obs = is_continuous(env, tuple_action, tuple_obs) |
| 447 | |
| 448 | n_actions_per_dim, action_ids, action_dim = action_stats( |
| 449 | env, md_action, cont_action, |
| 450 | ) |
| 451 | n_obs_per_dim, obs_ids, obs_dim = obs_stats(env, md_obs, cont_obs) |
| 452 | |
| 453 | env_info = { |
| 454 | "id": env.spec.id, |
| 455 | "seed": env.spec.seed if "seed" in dir(env.spec) else None, |
| 456 | "deterministic": bool(~env.spec.nondeterministic), |
| 457 | "tuple_actions": tuple_action, |
| 458 | "tuple_observations": tuple_obs, |
| 459 | "multidim_actions": md_action, |
| 460 | "multidim_observations": md_obs, |
| 461 | "continuous_actions": cont_action, |
| 462 | "continuous_observations": cont_obs, |
| 463 | "n_actions_per_dim": n_actions_per_dim, |
| 464 | "action_dim": action_dim, |
| 465 | "n_obs_per_dim": n_obs_per_dim, |
| 466 | "obs_dim": obs_dim, |
| 467 | "action_ids": action_ids, |
| 468 | "obs_ids": obs_ids, |
| 469 | } |
| 470 | |
| 471 | return env_info |
no test coverage detected