Restore Azure Tests (#3561)

mosaicml · Aug 20, 2024 · 4a5dfc1 · 4a5dfc1
1 parent f048254
commit 4a5dfc1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -182,6 +182,8 @@ filterwarnings = [
     '''ignore:.*The `NO_SHARD` sharding strategy is deprecated.*:FutureWarning''',
     # Some unit tests use ShardedTensor, though by default we use DTensor
     '''ignore:.*Please use DTensor instead and we are deprecating.*:FutureWarning''',
+    # Ignore NeptuneLogger warnings
+    '''ignore:.*NVML Shared Library Not Found. GPU usage metrics may not be reported.*''',
 ]
 
 # Coverage

diff --git a/tests/utils/object_store/test_azure_object_store.py b/tests/utils/object_store/test_azure_object_store.py
@@ -9,7 +9,6 @@
 
 
 @pytest.mark.remote
-@pytest.mark.skip(reason='Waiting for new Azure key to be approved')
 def test_azure_object_store_integration():
     model = SimpleModel()
     train_dataloader = DataLoader(dataset=RandomClassificationDataset())