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, step=0)[source]
Parameters:
Return type:

ndarray | None

check_ctrl(ctrl, step=0)[source]
Parameters:
Return type:

ndarray | None

dump(nan_env_ids, model_file, step)[source]
Parameters:
Return type:

str | None