Re-enabled enterprise tensorflow testing

activeloopai · Oct 10, 2023 · 06ebe3f · 06ebe3f
1 parent cc5bd17
commit 06ebe3f
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/deeplake/enterprise/test_tensorflow.py b/deeplake/enterprise/test_tensorflow.py
@@ -53,7 +53,6 @@ def index_transform(sample):
 @requires_libdeeplake
 @pytest.mark.slow
 @pytest.mark.flaky
-@pytest.mark.skip("causing lockups")
 def test_tensorflow_small(local_auth_ds):
     with local_auth_ds as ds:
         ds.create_tensor("image", max_chunk_size=TF_TESTS_MAX_CHUNK_SIZE)
@@ -119,7 +118,6 @@ def test_tensorflow_small(local_auth_ds):
 @requires_libdeeplake
 @pytest.mark.slow
 @pytest.mark.flaky
-@pytest.mark.skip("causing lockups")
 def test_tensorflow_transform(local_auth_ds):
     with local_auth_ds as ds:
         ds.create_tensor("image", max_chunk_size=TF_TESTS_MAX_CHUNK_SIZE)
@@ -277,7 +275,6 @@ def test_custom_tensor_order(local_auth_ds):
 @requires_libdeeplake
 @pytest.mark.slow
 @pytest.mark.flaky
-@pytest.mark.skip("causing lockups")
 def test_readonly_with_two_workers(local_auth_ds):
     local_auth_ds.create_tensor("images", max_chunk_size=TF_TESTS_MAX_CHUNK_SIZE)
     local_auth_ds.create_tensor("labels", max_chunk_size=TF_TESTS_MAX_CHUNK_SIZE)
@@ -544,7 +541,7 @@ def test_rename(local_auth_ds):
     "num_workers",
     [
         0,
-        pytest.param(2, marks=pytest.mark.skip("causing lockups")),
+        2,
     ],
 )
 @pytest.mark.slow
@@ -574,7 +571,7 @@ def test_indexes(local_auth_ds, num_workers):
     "num_workers",
     [
         0,
-        pytest.param(2, marks=pytest.mark.skip("causing lockups")),
+        2,
     ],
 )
 @pytest.mark.slow
@@ -646,7 +643,7 @@ def test_indexes_transform_dict(local_auth_ds, num_workers):
 @requires_tensorflow
 @requires_libdeeplake
 @pytest.mark.parametrize(
-    "num_workers", [0, pytest.param(2, marks=pytest.mark.skip("causing lockups"))]
+    "num_workers", [0, 2]
 )
 @pytest.mark.slow
 @pytest.mark.flaky