diff --git a/requirements/fabric/test.txt b/requirements/fabric/test.txt index 34571a9ee977e..9035164ea1462 100644 --- a/requirements/fabric/test.txt +++ b/requirements/fabric/test.txt @@ -1,4 +1,4 @@ -coverage ==7.8.0 +coverage ==7.8.2 numpy >=1.17.2, <1.27.0 pytest ==8.3.5 pytest-cov ==6.1.1 diff --git a/requirements/pytorch/test.txt b/requirements/pytorch/test.txt index e27b8aa33b3ea..fd4237ef74e66 100644 --- a/requirements/pytorch/test.txt +++ b/requirements/pytorch/test.txt @@ -1,4 +1,4 @@ -coverage ==7.8.0 +coverage ==7.8.2 pytest ==8.3.5 pytest-cov ==6.1.1 pytest-timeout ==2.4.0 diff --git a/tests/tests_pytorch/models/test_torchscript.py b/tests/tests_pytorch/models/test_torchscript.py index 8f9151265d21a..10a19974971eb 100644 --- a/tests/tests_pytorch/models/test_torchscript.py +++ b/tests/tests_pytorch/models/test_torchscript.py @@ -44,7 +44,9 @@ def test_torchscript_input_output(modelclass): model_output = model(model.example_input_array) script_output = script(model.example_input_array) - assert torch.allclose(script_output, model_output) + assert torch.allclose(script_output, model_output, rtol=1e-5, atol=1e-8), ( + f"Scripted output {script_output} does not match model output {model_output}." + ) @pytest.mark.skipif(_IS_WINDOWS and _TORCH_GREATER_EQUAL_2_4, reason="not close on Windows + PyTorch 2.4")