fix

Author: jafermarq

datasets/flwr_datasets/federated_dataset_test.py

@@ -629,14 +629,18 @@ def datasets_are_equal(ds1: Dataset, ds2: Dataset) -> bool:
         for key in row1:
             if key == "audio":
                 # Special handling for 'audio' key
-                if not all(
-                    [
-                        np.array_equal(row1[key]["array"], row2[key]["array"]),
-                        row1[key]["path"] == row2[key]["path"],
-                        row1[key]["sampling_rate"] == row2[key]["sampling_rate"],
-                    ]
-                ):
+                # Check array and sampling_rate
+                if not np.array_equal(row1[key]["array"], row2[key]["array"]):
                     return False
+                if row1[key]["sampling_rate"] != row2[key]["sampling_rate"]:
+                    return False
+
+                # Check path if available (AudioDecoder raises TypeError)
+                try:
+                    if row1[key]["path"] != row2[key]["path"]:
+                        return False
+                except TypeError:
+                    pass
             elif row1[key] != row2[key]:
                 # Direct comparison for other keys
                 return False