From 0e402e1478ff8c9db876eb7b3a708c1477cec2fa Mon Sep 17 00:00:00 2001 From: Sergii Dymchenko Date: Wed, 8 Nov 2023 11:58:36 -0800 Subject: [PATCH] Update deprecated `torch.range` in `test_modeling_ibert.py` (#27355) * Update deprecated torch.range * Remove comment --- tests/models/ibert/test_modeling_ibert.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/models/ibert/test_modeling_ibert.py b/tests/models/ibert/test_modeling_ibert.py index 096a55169a..b552cb75a5 100644 --- a/tests/models/ibert/test_modeling_ibert.py +++ b/tests/models/ibert/test_modeling_ibert.py @@ -519,7 +519,7 @@ class IBertModelIntegrationTest(unittest.TestCase): gelu_q = IntGELU(quant_mode=True) gelu_dq = nn.GELU() - x_int = torch.range(-10000, 10000, 1) + x_int = torch.arange(-10000, 10001, 1) x_scaling_factor = torch.tensor(0.001) x = x_int * x_scaling_factor @@ -534,7 +534,7 @@ class IBertModelIntegrationTest(unittest.TestCase): self.assertTrue(torch.allclose(q_int, q_int.round(), atol=1e-4)) def test_force_dequant_gelu(self): - x_int = torch.range(-10000, 10000, 1) + x_int = torch.arange(-10000, 10001, 1) x_scaling_factor = torch.tensor(0.001) x = x_int * x_scaling_factor @@ -565,7 +565,6 @@ class IBertModelIntegrationTest(unittest.TestCase): softmax_q = IntSoftmax(output_bit, quant_mode=True) softmax_dq = nn.Softmax() - # x_int = torch.range(-10000, 10000, 1) def _test(array): x_int = torch.tensor(array) x_scaling_factor = torch.tensor(0.1)