removing support to mps for torch_audiomentations

sensein · Oct 1, 2024 · 3a3b187 · 3a3b187
1 parent 1cdb2d9
commit 3a3b187
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -27,7 +27,7 @@ packages = [{include = "senselab", from = "src"}]
 
 [tool.poetry.dependencies]
 python = "~=3.10"
-datasets = "*"
+datasets = "~=3"
 torch = "~=2.4"
 torchvision = "~=0.19"
 torchaudio = "~=2.4"

diff --git a/src/senselab/audio/tasks/data_augmentation/torch_audiomentations.py b/src/senselab/audio/tasks/data_augmentation/torch_audiomentations.py
@@ -29,7 +29,7 @@ def augment_audios_with_torch_audiomentations(
         augmentation: A Composition of augmentations to run on each audio (uses torch-audiomentations), should have its
             output_type set to "dict"
         device: The device to use for augmenting. If the chosen device
-            is MPS or CUDA then the audios are all batched together, so for optimal performance, batching should
+            is CUDA then the audios are all batched together, so for optimal performance, batching should
             be done by passing a batch_size worth of audios ar a time.
             Default is None, which will select the device automatically.
 
@@ -40,7 +40,7 @@ def augment_audios_with_torch_audiomentations(
     """
     augmentation.output_type = "dict"
     device_type, dtype = _select_device_and_dtype(
-        user_preference=device, compatible_devices=[DeviceType.CUDA, DeviceType.MPS, DeviceType.CPU]
+        user_preference=device, compatible_devices=[DeviceType.CUDA, DeviceType.CPU]
     )
     if device_type == DeviceType.CPU: