admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-16 09:09:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2025-07-21 08:06:31 +00:00
committed by Adil Zouitine
parent 35612c61e1
commit 14c2ece004
5 changed files with 52 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/processor/test_pipeline.py
+6 -6
View File
@@ -902,17 +902,17 @@ class MockStepWithNonSerializableParam:
return (obs, action, reward, done, truncated, info, comp_data)
def get_config(self) -> Dict[str, Any]:
def get_config(self) -> dict[str, Any]:
# Note: env is intentionally NOT included here as it's not serializable
return {
"name": self.name,
"multiplier": self.multiplier,
}
def state_dict(self) -> Dict[str, torch.Tensor]:
def state_dict(self) -> dict[str, torch.Tensor]:
return {}
def load_state_dict(self, state: Dict[str, torch.Tensor]) -> None:
def load_state_dict(self, state: dict[str, torch.Tensor]) -> None:
pass
def reset(self) -> None:
@@ -936,16 +936,16 @@ class RegisteredMockStep:
return (obs, action, reward, done, truncated, info, comp_data)
def get_config(self) -> Dict[str, Any]:
def get_config(self) -> dict[str, Any]:
return {
"value": self.value,
"device": self.device,
}
def state_dict(self) -> Dict[str, torch.Tensor]:
def state_dict(self) -> dict[str, torch.Tensor]:
return {}
def load_state_dict(self, state: Dict[str, torch.Tensor]) -> None:
def load_state_dict(self, state: dict[str, torch.Tensor]) -> None:
pass
def reset(self) -> None: