admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 02:06:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(mypy): fixing mypy tests

Browse Source
This commit is contained in:
CarolinePascal
2026-01-09 18:29:16 +01:00
parent ab4903e752
commit 8b9451b585
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/configs/policies.py
+2
View File
@@ -153,6 +153,8 @@ class PreTrainedConfig(draccus.ChoiceRegistry, HubMixin, abc.ABC): # type: igno
@property @property
def audio_features(self) -> dict[str, PolicyFeature]: def audio_features(self) -> dict[str, PolicyFeature]:
if not self.input_features:
return {}
return {key: ft for key, ft in self.input_features.items() if ft.type is FeatureType.AUDIO} return {key: ft for key, ft in self.input_features.items() if ft.type is FeatureType.AUDIO}
@property @property