Merge pull request #1077 from activeloopai/fix/disable-hub-cloud-tests

[small] fix hub cloud throttling for tests
activeloopai · Jul 21, 2021 · 2047f2a · 2047f2a
2 parents bffb083 + c23e440
commit 2047f2a
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 5 deletions.
diff --git a/hub/api/tests/test_api.py b/hub/api/tests/test_api.py
@@ -21,7 +21,16 @@
 MAX_FLOAT_DTYPE = np.float_.__name__
 
 
-@enabled_persistent_dataset_generators
+# not using the predefined parametrizes because `hub_cloud_ds_generator` is not enabled by default
+@pytest.mark.parametrize(
+    "ds_generator",
+    [
+        "local_ds_generator",
+        "s3_ds_generator",
+        "hub_cloud_ds_generator",
+    ],
+    indirect=True,
+)
 def test_persist(ds_generator):
     ds = ds_generator()
 

diff --git a/hub/tests/dataset_fixtures.py b/hub/tests/dataset_fixtures.py
@@ -4,13 +4,13 @@
 
 enabled_datasets = pytest.mark.parametrize(
     "ds",
-    ["memory_ds", "local_ds", "s3_ds", "hub_cloud_ds"],
+    ["memory_ds", "local_ds", "s3_ds"],
     indirect=True,
 )
 
 enabled_persistent_dataset_generators = pytest.mark.parametrize(
     "ds_generator",
-    ["local_ds_generator", "s3_ds_generator", "hub_cloud_ds_generator"],
+    ["local_ds_generator", "s3_ds_generator"],
     indirect=True,
 )
 

diff --git a/hub/tests/storage_fixtures.py b/hub/tests/storage_fixtures.py
@@ -7,13 +7,13 @@
 
 enabled_storages = pytest.mark.parametrize(
     "storage",
-    ["memory_storage", "local_storage", "s3_storage", "hub_cloud_storage"],
+    ["memory_storage", "local_storage", "s3_storage"],
     indirect=True,
 )
 
 enabled_persistent_storages = pytest.mark.parametrize(
     "storage",
-    ["local_storage", "s3_storage", "hub_cloud_storage"],
+    ["local_storage", "s3_storage"],
     indirect=True,
 )