From 8b9451b5855993e6c5e3f059eef2879a6e205eb8 Mon Sep 17 00:00:00 2001
From: CarolinePascal <caroline8.pascal@gmail.com>
Date: Fri, 9 Jan 2026 18:29:16 +0100
Subject: [PATCH] fix(mypy): fixing mypy tests

---
 src/lerobot/configs/policies.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/lerobot/configs/policies.py b/src/lerobot/configs/policies.py
index 11494f0fd..476e70c92 100644
--- a/src/lerobot/configs/policies.py
+++ b/src/lerobot/configs/policies.py
@@ -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