Merge pull request #70 from ahmedshahabaz/team2

fix the find_audio_features function to search for .pt files
sensein · Jun 27, 2024 · dd025ca · dd025ca
2 parents 5c270d3 + 56fb530
commit dd025ca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/b2aiprep/dataset.py b/src/b2aiprep/dataset.py
@@ -332,9 +332,9 @@ def find_audio_features(self, subject_id: str, session_id: str) -> t.List[Path]:
         List[Path]
             A list of audio features.
         """
-        session_path = self.data_path / f"sub-{subject_id}" / f"ses-{session_id}" / "audio_features"
+        session_path = self.data_path / f"sub-{subject_id}" / f"ses-{session_id}" / "audio"
         features = []
-        for feature_file in session_path.glob("*.json"):
+        for feature_file in session_path.glob("*.pt"):
             features.append(feature_file)
         return features