Skip to content

Commit 2b79aa9

Browse files
committed
increase testing tolerence
Signed-off-by: Farhan Ahmed <[email protected]>
1 parent 66e0e5a commit 2b79aa9

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

tests/estimators/object_detection/test_pytorch_detection_transformer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ def test_loss_gradient(art_warning, get_pytorch_detr):
125125
-0.00526983,
126126
]
127127
)
128-
np.testing.assert_array_almost_equal(grads[0, 0, 208, 192:224], expected_gradients1, decimal=1)
128+
np.testing.assert_array_almost_equal(grads[0, 0, 208, 192:224], expected_gradients1, decimal=2)
129129

130130
expected_gradients2 = np.asarray(
131131
[
@@ -163,7 +163,7 @@ def test_loss_gradient(art_warning, get_pytorch_detr):
163163
0.00305779,
164164
]
165165
)
166-
np.testing.assert_array_almost_equal(grads[0, 0, 192:224, 208], expected_gradients2, decimal=1)
166+
np.testing.assert_array_almost_equal(grads[0, 0, 192:224, 208], expected_gradients2, decimal=2)
167167

168168
except ARTTestException as e:
169169
art_warning(e)

tests/estimators/object_detection/test_pytorch_yolo.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ def test_loss_gradient(art_warning, get_pytorch_yolo):
119119
0.00232432,
120120
]
121121
)
122-
np.testing.assert_array_almost_equal(grads[0, 0, 208, 192:224], expected_gradients1, decimal=1)
122+
np.testing.assert_array_almost_equal(grads[0, 0, 208, 192:224], expected_gradients1, decimal=2)
123123

124124
expected_gradients2 = np.asarray(
125125
[
@@ -157,7 +157,7 @@ def test_loss_gradient(art_warning, get_pytorch_yolo):
157157
0.00860081,
158158
]
159159
)
160-
np.testing.assert_array_almost_equal(grads[0, 0, 192:224, 208], expected_gradients2, decimal=1)
160+
np.testing.assert_array_almost_equal(grads[0, 0, 192:224, 208], expected_gradients2, decimal=2)
161161

162162
except ARTTestException as e:
163163
art_warning(e)

0 commit comments

Comments
 (0)