Fixed floor_divide deprecation warnings seen in pytest output (pytorc…

…h#1455) * Fixed floor_divide deprecation warnings seen in pytest output * Fixed warning in test_flanger_triangle_linear
carolineechen · Apr 30, 2021 · fcfa07a · fcfa07a
1 parent 95ab342
commit fcfa07a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/test/torchaudio_unittest/functional/functional_cpu_test.py b/test/torchaudio_unittest/functional/functional_cpu_test.py
@@ -199,7 +199,8 @@ def test_mask_along_axis(self, shape, mask_param, mask_value, axis):
 
         masked_columns = (mask_specgram == mask_value).sum(other_axis)
         num_masked_columns = (masked_columns == mask_specgram.size(other_axis)).sum()
-        num_masked_columns //= mask_specgram.size(0)
+        num_masked_columns = torch.div(
+            num_masked_columns, mask_specgram.size(0), rounding_mode='floor')
 
         assert mask_specgram.size() == specgram.size()
         assert num_masked_columns < mask_param

diff --git a/torchaudio/functional/filtering.py b/torchaudio/functional/filtering.py
@@ -46,7 +46,7 @@ def _generate_wave_table(
         d = (torch.sin(point.to(torch.float64) / table_size * 2 * math.pi) + 1) / 2
     elif wave_type == "TRIANGLE":
         d = point.to(torch.float64) * 2 / table_size
-        value = 4 * point // table_size
+        value = torch.div(4 * point, table_size, rounding_mode='floor')
         d[value == 0] = d[value == 0] + 0.5
         d[value == 1] = 1.5 - d[value == 1]
         d[value == 2] = 1.5 - d[value == 2]