Fix type annotations: replace builtin any with typing.Any

Occurrences in _internal.py, config.py, dbutils.py, oauth.py, and
runtime/dbutils_stub.py were using the builtin any (an Iterable->bool
callable) as a type annotation instead of typing.Any. mypy interprets
these as the callable type, causing spurious attr-defined errors on
annotated values.

Signed-off-by: Victor Hugo Xavier <victor@coactive.ai>
Signed-off-by: coactive-victor <victor@coactive.ai>

Author: coactive-victor

databricks/sdk/config.py

@@ -6,21 +6,23 @@
 import pathlib
 import re
 import urllib.parse
-from typing import Dict, Iterable, List, Optional
+from typing import Any, Dict, Iterable, List, Optional
 
 import requests
 
 from . import useragent
 from ._base_client import _fix_host_if_needed
 from .client_types import ClientType, HostType
 from .clock import Clock, RealClock
-from .credentials_provider import (CredentialsStrategy, DefaultCredentials,
-                                   OAuthCredentialsProvider)
-from .environments import (ALL_ENVS, AzureEnvironment, Cloud,
-                           DatabricksEnvironment, get_environment_for_hostname)
-from .oauth import (OidcEndpoints, Token,
-                    get_azure_entra_id_workspace_endpoints,
-                    get_endpoints_from_url, get_host_metadata)
+from .credentials_provider import CredentialsStrategy, DefaultCredentials, OAuthCredentialsProvider
+from .environments import ALL_ENVS, AzureEnvironment, Cloud, DatabricksEnvironment, get_environment_for_hostname
+from .oauth import (
+    OidcEndpoints,
+    Token,
+    get_azure_entra_id_workspace_endpoints,
+    get_endpoints_from_url,
+    get_host_metadata,
+)
 
 logger = logging.getLogger("databricks.sdk")
 
@@ -47,7 +49,7 @@ def __get__(self, cfg: "Config", owner):
             return None
         return cfg._inner.get(self.name, None)
 
-    def __set__(self, cfg: "Config", value: any):
+    def __set__(self, cfg: "Config", value: Any):
         cfg._inner[self.name] = self.transform(value)
 
     def __repr__(self) -> str:

databricks/sdk/dbutils.py

@@ -186,8 +186,8 @@ def get(
             self,
             taskKey: str,
             key: str,
-            default: any = None,
-            debugValue: any = None,
+            default: Any = None,
+            debugValue: Any = None,
         ) -> None:
             """
             Returns `debugValue` if present, throws an error otherwise as this implementation is always run outside of a job run
@@ -198,7 +198,7 @@ def get(
                 )
             return debugValue
 
-        def set(self, key: str, value: any) -> None:
+        def set(self, key: str, value: Any) -> None:
             """
             Sets a task value on the current task run
             """

databricks/sdk/oauth.py

@@ -789,7 +789,7 @@ def from_host(
         from .credentials_provider import credentials_strategy
 
         @credentials_strategy("noop", [])
-        def noop_credentials(_: any):
+        def noop_credentials(_: Any):
             return lambda: {}
 
         config = Config(host=host, credentials_strategy=noop_credentials)

databricks/sdk/runtime/dbutils_stub.py

@@ -57,7 +57,7 @@ class data:
         """
 
         @staticmethod
-        def summarize(df: any, precise: bool = False) -> None:
+        def summarize(df: typing.Any, precise: bool = False) -> None:
             """Summarize a Spark/pandas/Koalas DataFrame and visualize the statistics to get quick insights.
 
             Example: dbutils.data.summarize(df)
@@ -200,16 +200,16 @@ class taskValues:
             def get(
                 taskKey: str,
                 key: str,
-                default: any = None,
-                debugValue: any = None,
+                default: typing.Any = None,
+                debugValue: typing.Any = None,
             ) -> None:
                 """
                 Returns the latest task value that belongs to the current job run
                 """
                 ...
 
             @staticmethod
-            def set(key: str, value: any) -> None:
+            def set(key: str, value: typing.Any) -> None:
                 """
                 Sets a task value on the current task run
                 """

databricks/sdk/service/_internal.py

@@ -8,13 +8,13 @@
 from databricks.sdk.common.types.fieldmask import FieldMask
 
 
-def _from_dict(d: Dict[str, any], field: str, cls: Type) -> any:
+def _from_dict(d: Dict[str, Any], field: str, cls: Type) -> Any:
     if field not in d or d[field] is None:
         return None
     return getattr(cls, "from_dict")(d[field])
 
 
-def _repeated_dict(d: Dict[str, any], field: str, cls: Type) -> any:
+def _repeated_dict(d: Dict[str, Any], field: str, cls: Type) -> Any:
     if field not in d or not d[field]:
         return []
     from_dict = getattr(cls, "from_dict")
@@ -28,14 +28,14 @@ def _get_enum_value(cls: Type, value: str) -> Optional[Type]:
     )
 
 
-def _enum(d: Dict[str, any], field: str, cls: Type) -> any:
+def _enum(d: Dict[str, Any], field: str, cls: Type) -> Any:
     """Unknown enum values are returned as None."""
     if field not in d or not d[field]:
         return None
     return _get_enum_value(cls, d[field])
 
 
-def _repeated_enum(d: Dict[str, any], field: str, cls: Type) -> any:
+def _repeated_enum(d: Dict[str, Any], field: str, cls: Type) -> Any:
     """For now, unknown enum values are not included in the response."""
     if field not in d or not d[field]:
         return None