From 093a45765695dafdca14df702e5f5f564f9e46bd Mon Sep 17 00:00:00 2001 From: activesoull Date: Tue, 27 Aug 2024 18:51:59 +0400 Subject: [PATCH] mark test to rerun in case if the responses is not 200 --- deeplake/api/tests/test_access_method.py | 1 + deeplake/core/vectorstore/test_deeplake_vectorstore.py | 4 ++++ 2 files changed, 5 insertions(+) diff --git a/deeplake/api/tests/test_access_method.py b/deeplake/api/tests/test_access_method.py index ac05f60468..e08251499e 100644 --- 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 index c3af4c549a..811e231614 100644 --- 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,