tests: flaky `test_servable_module_validator` (#17985)

2023-07-04 16:04:16 +02:00 · 2023-07-04 16:04:16 +02:00 · e49a5b6662
parent 036bd74e47
commit e49a5b6662
1 changed files with 1 additions and 0 deletions
--- 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()