Merge pull request #1680 from davidslater/mp3-compression-bug

Mp3Compression defense bug fix - handle case when input is 2D but not an object array

Author: beat-buesser

art/defences/preprocessor/mp3_compression.py

@@ -115,6 +115,10 @@ def wav_to_mp3(x, sample_rate):
                 x_mp3 = x_mp3 * 2 ** -15
             return x_mp3.astype(x_dtype)
 
+        x_orig_type = x.dtype
+        if x.dtype != object and x.ndim == 2:
+            x = x.astype(object)
+
         if x.dtype != object and x.ndim != 3:
             raise ValueError("Mp3 compression can only be applied to temporal data across at least one channel.")
 
@@ -146,6 +150,9 @@ def wav_to_mp3(x, sample_rate):
         if x.dtype != object and self.channels_first:
             x_mp3 = np.swapaxes(x_mp3, 1, 2)
 
+        if x_orig_type != object and x.dtype == object and x.ndim == 2:
+            x_mp3 = x_mp3.astype(x_orig_type)
+
         return x_mp3, y
 
     def _check_params(self) -> None: