mark test to rerun in case if the responses is not 200

activeloopai · Aug 27, 2024 · 093a457 · 093a457
1 parent fbb5c82
commit 093a457
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/deeplake/api/tests/test_access_method.py b/deeplake/api/tests/test_access_method.py
@@ -81,6 +81,7 @@ def test_access_method(s3_ds_generator):
 
 
 @pytest.mark.slow
+@pytest.mark.flaky(reruns=3)
 def test_access_method_with_creds(
     hub_cloud_ds_generator, hub_cloud_dev_managed_creds_key
 ):

diff --git a/deeplake/core/vectorstore/test_deeplake_vectorstore.py b/deeplake/core/vectorstore/test_deeplake_vectorstore.py
@@ -958,6 +958,7 @@ def assert_updated_vector_store(
 )
 @pytest.mark.parametrize("init_embedding_function", [embedding_fn3, None])
 @pytest.mark.slow
+@pytest.mark.flaky(reruns=3)
 @requires_libdeeplake
 def test_update_embedding(
     ds,
@@ -1344,6 +1345,7 @@ def create_and_populate_vs(
 
 
 @requires_libdeeplake
+@pytest.mark.flaky(reruns=3)
 def test_update_embedding_row_ids_and_ids_specified_should_throw_exception(
     local_path,
     vector_store_hash_ids,
@@ -1368,6 +1370,7 @@ def test_update_embedding_row_ids_and_ids_specified_should_throw_exception(
 
 
 @requires_libdeeplake
+@pytest.mark.flaky(reruns=3)
 def test_update_embedding_row_ids_and_filter_specified_should_throw_exception(
     local_path,
     vector_store_filters,
@@ -1391,6 +1394,7 @@ def test_update_embedding_row_ids_and_filter_specified_should_throw_exception(
 
 
 @requires_libdeeplake
+@pytest.mark.flaky(reruns=3)
 def test_update_embedding_query_and_filter_specified_should_throw_exception(
     local_path,
     vector_store_filters,