unilab.utils.nan_guard
NaN/Inf guard for env-layer numerical anomaly detection and state dumping.
Classes
NanGuard
|
|
NanGuardCfg
|
NanGuardCfg(enabled: 'bool' = False, buffer_size: 'int' = 100, max_envs_to_dump: 'int' = 5, output_dir: 'str | None' = None) |
-
class unilab.utils.nan_guard.NanGuardCfg[source]
Bases: object
NanGuardCfg(enabled: ‘bool’ = False, buffer_size: ‘int’ = 100, max_envs_to_dump: ‘int’ = 5, output_dir: ‘str | None’ = None)
- Parameters:
-
-
enabled:
bool = False
-
buffer_size:
int = 100
-
max_envs_to_dump:
int = 5
-
output_dir:
str | None = None
-
__init__(enabled=False, buffer_size=100, max_envs_to_dump=5, output_dir=None)
- Parameters:
-
-
class unilab.utils.nan_guard.NanGuard[source]
Bases: object
- Parameters:
-
-
__init__(cfg, num_envs, supports_state_playback)[source]
- Parameters:
-
-
capture(physics_state)[source]
- Parameters:
physics_state (ndarray | None)
- Return type:
None
-
check(obs, reward)[source]
- Parameters:
-
- Return type:
ndarray | None
-
dump(nan_env_ids, model_file, step)[source]
- Parameters:
-
- Return type:
str | None