mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-24 18:26:11 +00:00
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
@@ -272,6 +272,7 @@ class HILEnvConfig(EnvConfig):
|
|||||||
"gripper_penalty": self.gripper_penalty,
|
"gripper_penalty": self.gripper_penalty,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@EnvConfig.register_subclass("libero")
|
@EnvConfig.register_subclass("libero")
|
||||||
@dataclass
|
@dataclass
|
||||||
class LiberoEnv(EnvConfig):
|
class LiberoEnv(EnvConfig):
|
||||||
|
|||||||
@@ -2,7 +2,8 @@ import math
|
|||||||
import os
|
import os
|
||||||
from collections import defaultdict
|
from collections import defaultdict
|
||||||
from itertools import chain
|
from itertools import chain
|
||||||
from typing import Any, Callable
|
from typing import Any
|
||||||
|
from collections.abc import Callable
|
||||||
|
|
||||||
import gymnasium as gym
|
import gymnasium as gym
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
|||||||
@@ -55,13 +55,12 @@ from copy import deepcopy
|
|||||||
from dataclasses import asdict
|
from dataclasses import asdict
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from pprint import pformat
|
from pprint import pformat
|
||||||
from typing import Callable
|
from collections.abc import Callable
|
||||||
|
|
||||||
import einops
|
import einops
|
||||||
import gymnasium as gym
|
import gymnasium as gym
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import torch
|
import torch
|
||||||
from termcolor import colored
|
|
||||||
from torch import Tensor, nn
|
from torch import Tensor, nn
|
||||||
from tqdm import trange
|
from tqdm import trange
|
||||||
|
|
||||||
@@ -73,7 +72,6 @@ from lerobot.policies.factory import make_policy
|
|||||||
from lerobot.policies.pretrained import PreTrainedPolicy
|
from lerobot.policies.pretrained import PreTrainedPolicy
|
||||||
from lerobot.policies.utils import get_device_from_parameters
|
from lerobot.policies.utils import get_device_from_parameters
|
||||||
from lerobot.utils.io_utils import write_video
|
from lerobot.utils.io_utils import write_video
|
||||||
from lerobot.utils.random_utils import set_seed
|
|
||||||
from lerobot.utils.utils import (
|
from lerobot.utils.utils import (
|
||||||
get_safe_torch_device,
|
get_safe_torch_device,
|
||||||
init_logging,
|
init_logging,
|
||||||
@@ -456,6 +454,7 @@ def _compile_episode_data(
|
|||||||
|
|
||||||
return data_dict
|
return data_dict
|
||||||
|
|
||||||
|
|
||||||
@parser.wrap()
|
@parser.wrap()
|
||||||
def eval(cfg: EvalPipelineConfig):
|
def eval(cfg: EvalPipelineConfig):
|
||||||
logging.info(pformat(asdict(cfg)))
|
logging.info(pformat(asdict(cfg)))
|
||||||
@@ -523,6 +522,7 @@ def eval(cfg: EvalPipelineConfig):
|
|||||||
|
|
||||||
logging.info("End of eval")
|
logging.info("End of eval")
|
||||||
|
|
||||||
|
|
||||||
def eval_policy_multitask(
|
def eval_policy_multitask(
|
||||||
envs: dict[str, dict[str, gym.vector.VectorEnv]],
|
envs: dict[str, dict[str, gym.vector.VectorEnv]],
|
||||||
policy,
|
policy,
|
||||||
@@ -545,7 +545,14 @@ def eval_policy_multitask(
|
|||||||
"""Evaluates a single task in parallel."""
|
"""Evaluates a single task in parallel."""
|
||||||
print(f"Evaluating: task_group: {task_group}, task_id: {task_id} ...")
|
print(f"Evaluating: task_group: {task_group}, task_id: {task_id} ...")
|
||||||
task_result = eval_policy(
|
task_result = eval_policy(
|
||||||
env, policy, n_episodes, max_episodes_rendered, videos_dir, return_episode_data, start_seed, verbose=verbose
|
env,
|
||||||
|
policy,
|
||||||
|
n_episodes,
|
||||||
|
max_episodes_rendered,
|
||||||
|
videos_dir,
|
||||||
|
return_episode_data,
|
||||||
|
start_seed,
|
||||||
|
verbose=verbose,
|
||||||
)
|
)
|
||||||
|
|
||||||
per_episode = task_result["per_episode"]
|
per_episode = task_result["per_episode"]
|
||||||
@@ -629,6 +636,7 @@ def eval_policy_multitask(
|
|||||||
|
|
||||||
return results
|
return results
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
init_logging()
|
init_logging()
|
||||||
eval_main()
|
eval_main()
|
||||||
|
|||||||
Reference in New Issue
Block a user