nvkartik commited on
Commit
174b9fb
·
1 Parent(s): 76ba10d

minor fix for state only envs

Browse files
Files changed (1) hide show
  1. env.py +2 -2
env.py CHANGED
@@ -428,8 +428,8 @@ def _create_isaaclab_env(config: dict, n_envs: int) -> dict[str, dict[int, gym.v
428
  logging.info(f"Set render_mode={render_mode} on underlying IsaacLab env")
429
 
430
  # Validate config
431
- state_keys = tuple(k.strip() for k in config.get("state_keys", "").split(",") if k.strip())
432
- camera_keys = tuple(k.strip() for k in config.get("camera_keys", "").split(",") if k.strip())
433
  validate_config(
434
  raw_env, state_keys, camera_keys,
435
  config.get("state_dim", 54),
 
428
  logging.info(f"Set render_mode={render_mode} on underlying IsaacLab env")
429
 
430
  # Validate config
431
+ state_keys = tuple(k.strip() for k in (config.get("state_keys") or "").split(",") if k.strip())
432
+ camera_keys = tuple(k.strip() for k in (config.get("camera_keys") or "").split(",") if k.strip())
433
  validate_config(
434
  raw_env, state_keys, camera_keys,
435
  config.get("state_dim", 54),