From e49a5b6662177815f4cfde464ea65f453ba3861d Mon Sep 17 00:00:00 2001
From: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Date: Tue, 4 Jul 2023 16:04:16 +0200
Subject: [PATCH] tests: flaky `test_servable_module_validator` (#17985)

---
 tests/tests_pytorch/serve/test_servable_module_validator.py | 1 +
 1 file changed, 1 insertion(+)

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()