diff --git a/tests/tests_pytorch/serve/test_servable_module_validator.py b/tests/tests_pytorch/serve/test_servable_module_validator.py index f1aba23146..2d7d56391c 100644 --- a/tests/tests_pytorch/serve/test_servable_module_validator.py +++ b/tests/tests_pytorch/serve/test_servable_module_validator.py @@ -30,6 +30,7 @@ class ServableBoringModel(BoringModel, ServableModule): return {"output": [0, 1]} +@pytest.mark.flaky(reruns=3) def test_servable_module_validator(): model = ServableBoringModel() callback = ServableModuleValidator()