From 0c8059457fb4593c50afa520952c9fad0de54c61 Mon Sep 17 00:00:00 2001 From: Stuart Wheaton Date: Fri, 9 Feb 2024 15:14:08 -0500 Subject: [PATCH 1/2] add **kwargs to AzureStorageClient init --- eta/core/storage.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/eta/core/storage.py b/eta/core/storage.py index 3ec961b5..02e58162 100644 --- a/eta/core/storage.py +++ b/eta/core/storage.py @@ -2625,7 +2625,7 @@ class AzureStorageClient( strategy used by this class. """ - def __init__(self, credentials=None, max_pool_connections=None): + def __init__(self, credentials=None, max_pool_connections=None, **kwargs): """Creates an AzureStorageClient instance. Args: @@ -2633,6 +2633,9 @@ def __init__(self, credentials=None, max_pool_connections=None): automatically loaded as described in `NeedsAzureCredentials` max_pool_connections: an optional maximum number of connections to keep in the connection pool + **kwargs: optional configuration options for + `azure.identity.DefaultAzureCredential(**kwargs)` or + `azure.identity.ClientSecretCredential(**kwargs)` """ if credentials is None: credentials, _ = self.load_credentials() @@ -2685,10 +2688,10 @@ def __init__(self, credentials=None, max_pool_connections=None): credential = account_key elif tenant_id and client_id and client_secret: credential = azi.ClientSecretCredential( - tenant_id, client_id, client_secret + tenant_id, client_id, client_secret, **kwargs ) else: - credential = azi.DefaultAzureCredential() + credential = azi.DefaultAzureCredential(**kwargs) if account_url is None: if account_name is None: From 78c67ab0601fd31e7c63d5bc9cc99530571a9c0d Mon Sep 17 00:00:00 2001 From: Stuart Wheaton Date: Mon, 12 Feb 2024 09:11:10 -0500 Subject: [PATCH 2/2] bump --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 8c48de2f..38e4e253 100644 --- a/setup.py +++ b/setup.py @@ -16,7 +16,7 @@ from wheel.bdist_wheel import bdist_wheel -VERSION = "0.12.4" +VERSION = "0.12.5" class BdistWheelCustom(bdist_wheel):