mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-24 04:59:47 +00:00
chore(style): pre-commit envs
This commit is contained in:
@@ -314,6 +314,7 @@ lerobot-eval \
|
|||||||
```
|
```
|
||||||
|
|
||||||
All `--env_kwargs.*` arguments will be collected into a dictionary and passed as keyword arguments to the hub environment's `make_env` function. This allows you to:
|
All `--env_kwargs.*` arguments will be collected into a dictionary and passed as keyword arguments to the hub environment's `make_env` function. This allows you to:
|
||||||
|
|
||||||
- Pass configuration file paths
|
- Pass configuration file paths
|
||||||
- Override default settings
|
- Override default settings
|
||||||
- Specify custom task parameters
|
- Specify custom task parameters
|
||||||
|
|||||||
@@ -155,7 +155,9 @@ def make_env(
|
|||||||
|
|
||||||
# call the hub-provided make_env with any additional kwargs
|
# call the hub-provided make_env with any additional kwargs
|
||||||
env_cfg = None if isinstance(cfg, str) else cfg
|
env_cfg = None if isinstance(cfg, str) else cfg
|
||||||
raw_result = _call_make_env(module, n_envs=n_envs, use_async_envs=use_async_envs, cfg=env_cfg, **kwargs)
|
raw_result = _call_make_env(
|
||||||
|
module, n_envs=n_envs, use_async_envs=use_async_envs, cfg=env_cfg, **kwargs
|
||||||
|
)
|
||||||
|
|
||||||
# normalize the return into {suite: {task_id: vec_env}}
|
# normalize the return into {suite: {task_id: vec_env}}
|
||||||
return _normalize_hub_result(raw_result)
|
return _normalize_hub_result(raw_result)
|
||||||
|
|||||||
@@ -532,7 +532,7 @@ def eval_main(cfg: EvalPipelineConfig):
|
|||||||
n_envs=cfg.eval.batch_size,
|
n_envs=cfg.eval.batch_size,
|
||||||
use_async_envs=cfg.eval.use_async_envs,
|
use_async_envs=cfg.eval.use_async_envs,
|
||||||
trust_remote_code=cfg.trust_remote_code,
|
trust_remote_code=cfg.trust_remote_code,
|
||||||
**cfg.env_kwargs
|
**cfg.env_kwargs,
|
||||||
)
|
)
|
||||||
|
|
||||||
logging.info("Making policy.")
|
logging.info("Making policy.")
|
||||||
|
|||||||
Reference in New Issue
Block a user