Fix unit test

Signed-off-by: Beat Buesser <[email protected]>
Trusted-AI · Dec 24, 2023 · ae6fe21 · ae6fe21
1 parent 11aac26
commit ae6fe21
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/estimators/object_detection/test_pytorch_detection_transformer.py b/tests/estimators/object_detection/test_pytorch_detection_transformer.py
@@ -145,6 +145,8 @@ def test_loss_gradient(art_warning, get_pytorch_detr):
 
         print("expected_gradients1")
         print(grads[0, 0, 10, :32])
+        print("expected_gradients2")
+        print(grads[1, 0, 10, :32])
 
         np.testing.assert_array_almost_equal(grads[0, 0, 10, :32], expected_gradients1, decimal=4)
 
@@ -185,9 +187,6 @@ def test_loss_gradient(art_warning, get_pytorch_detr):
             ]
         )
 
-        print("expected_gradients2")
-        print(grads[1, 0, 10, :32])
-
         np.testing.assert_array_almost_equal(grads[1, 0, 10, :32], expected_gradients2, decimal=4)
 
     except ARTTestException as e: