fix(mypy): fixing mypy tests

This commit is contained in:
CarolinePascal
2026-01-09 18:29:16 +01:00
parent ab4903e752
commit 8b9451b585
+2
View File
@@ -153,6 +153,8 @@ class PreTrainedConfig(draccus.ChoiceRegistry, HubMixin, abc.ABC): # type: igno
@property
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}
@property