mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-22 20:19:43 +00:00
refactor(envs): remove unused add_envs_task
Replaced by env.call("task_description") in lerobot_eval.py. No callers
remain in the codebase.
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -29,7 +29,6 @@ from torch import Tensor
|
|||||||
|
|
||||||
from lerobot.configs.types import FeatureType, PolicyFeature
|
from lerobot.configs.types import FeatureType, PolicyFeature
|
||||||
from lerobot.envs.configs import EnvConfig
|
from lerobot.envs.configs import EnvConfig
|
||||||
from lerobot.types import RobotObservation
|
|
||||||
from lerobot.utils.constants import OBS_ENV_STATE, OBS_IMAGE, OBS_IMAGES, OBS_STATE, OBS_STR
|
from lerobot.utils.constants import OBS_ENV_STATE, OBS_IMAGE, OBS_IMAGES, OBS_STATE, OBS_STR
|
||||||
from lerobot.utils.utils import get_channel_first_image_shape
|
from lerobot.utils.utils import get_channel_first_image_shape
|
||||||
|
|
||||||
@@ -206,28 +205,6 @@ def check_env_attributes_and_types(env: gym.vector.VectorEnv) -> None:
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def add_envs_task(env: gym.vector.VectorEnv, observation: RobotObservation) -> RobotObservation:
|
|
||||||
"""Adds task feature to the observation dict with respect to the first environment attribute."""
|
|
||||||
if _sub_env_has_attr(env, "task_description"):
|
|
||||||
task_result = list(env.call("task_description"))
|
|
||||||
|
|
||||||
if not all(isinstance(item, str) for item in task_result):
|
|
||||||
raise TypeError("All items in task_description result must be strings")
|
|
||||||
|
|
||||||
observation["task"] = task_result
|
|
||||||
elif _sub_env_has_attr(env, "task"):
|
|
||||||
task_result = list(env.call("task"))
|
|
||||||
|
|
||||||
if not all(isinstance(item, str) for item in task_result):
|
|
||||||
raise TypeError("All items in task result must be strings")
|
|
||||||
|
|
||||||
observation["task"] = task_result
|
|
||||||
else:
|
|
||||||
num_envs = observation[list(observation.keys())[0]].shape[0]
|
|
||||||
observation["task"] = ["" for _ in range(num_envs)]
|
|
||||||
return observation
|
|
||||||
|
|
||||||
|
|
||||||
def _close_single_env(env: Any) -> None:
|
def _close_single_env(env: Any) -> None:
|
||||||
try:
|
try:
|
||||||
env.close()
|
env.close()
|
||||||
|
|||||||
Reference in New Issue
Block a user