c

pola-rs · Jan 31, 2025 · dcc2d34 · dcc2d34
1 parent 61b6e1d
commit dcc2d34
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/py-polars/polars/catalog.py b/py-polars/polars/catalog.py
@@ -469,12 +469,14 @@ def _get_databricks_token(cls) -> str:
 
 
 class CatalogCredentialProvider:
-    def __init__(self, catalog: Catalog, table_id: str, *, write: bool):
+    """Retrieves credentials from the Unity catalog temporary credentials API."""
+
+    def __init__(self, catalog: Catalog, table_id: str, *, write: bool) -> None:
         self.catalog = catalog
         self.table_id = table_id
         self.write = write
 
-    def __call__(self) -> CredentialProviderFunctionReturn:
+    def __call__(self) -> CredentialProviderFunctionReturn:  # noqa: D102
         creds, _, expiry = self.catalog._get_table_credentials(
             self.table_id, write=self.write
         )