updated docs

Author: jmmshn-ms

src/maggma/stores/azure.py

@@ -41,10 +41,11 @@
 ]
 
 
-def _get_azure_credential(credential_class: str):
+def _get_azure_credential(credential_class):
     """Import the azure.identity module and return the credential class."""
-    module_name = "azure.identity"
-    credential_class = getattr(importlib.import_module(module_name), credential_class)
+    if isinstance(credential_class, str):
+        module_name = "azure.identity"
+        credential_class = getattr(importlib.import_module(module_name), credential_class)
     return credential_class()
 
 
@@ -86,7 +87,9 @@ def __init__(
                 Currently supported keywords:
                     - connection_string: a connection string for the Azure blob
             credential_type: the type of credential to use to authenticate with Azure.
-                Default is "DefaultAzureCredential".
+                Default is "DefaultAzureCredential".  For serializable stores, provide
+                a string representation of the credential class. Otherwises, you may
+                provide the class itself.
             compress: compress files inserted into the store
             sub_dir: (optional)  subdirectory of the container to store the data.
                 When defined, a final "/" will be added if not already present.

tests/stores/test_azure.py

@@ -422,19 +422,32 @@ def test_no_login():
         store.connect()
 
 
-@pytest.mark.parametrize(
-    "credential_type",
-    [
-        "DefaultAzureCredential",
-        "AzureCliCredential",
-    ],
-)
-def test_credential_type_valid(credential_type):
-    with azurite_container():
-        index = MemoryStore("index")
-        store = AzureBlobStore(
-            index,
-            AZURITE_CONTAINER_NAME,
-            credential_type=credential_type,
-        )
-        store.connect()
+def test_credential_type_valid():
+    credential_type = "DefaultAzureCredential"
+    index = MemoryStore("index")
+    store = AzureBlobStore(
+        index,
+        AZURITE_CONTAINER_NAME,
+        credential_type=credential_type,
+    )
+    assert store.credential_type == credential_type
+
+    credential_type = "AzureCliCredential"
+    index = MemoryStore("index")
+    store = AzureBlobStore(
+        index,
+        AZURITE_CONTAINER_NAME,
+        credential_type=credential_type,
+    )
+    assert store.credential_type == credential_type
+
+    from azure.identity import DefaultAzureCredential
+
+    credential_type = DefaultAzureCredential
+    index = MemoryStore("index")
+    store = AzureBlobStore(
+        index,
+        AZURITE_CONTAINER_NAME,
+        credential_type=credential_type,
+    )
+    assert not isinstance(store.credential_type, str)