Merge pull request #336 from qwaker00/patch-1

Don't calculate heavy room parameters if augmentation is not going to be applied
iver56 · Apr 29, 2024 · fe04ca2 · fe04ca2
2 parents 956e3ca + dd6cf05
commit fe04ca2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/audiomentations/augmentations/room_simulator.py b/audiomentations/augmentations/room_simulator.py
@@ -204,6 +204,9 @@ def randomize_parameters(self, samples: NDArray[np.float32], sample_rate: int):
             raise
 
         super().randomize_parameters(samples, sample_rate)
+        if not self.parameters["should_apply"]:
+            return
+
         self.parameters["size_x"] = random.uniform(self.min_size_x, self.max_size_x)
         self.parameters["size_y"] = random.uniform(self.min_size_y, self.max_size_y)
         self.parameters["size_z"] = random.uniform(self.min_size_z, self.max_size_z)