Correct use of any as a type to typing.Any

databricks/sdk/config.py

@@ -6,7 +6,7 @@
 import pathlib
 import re
 import urllib.parse
-from typing import Dict, Iterable, List, Optional
+from typing import Any, Dict, Iterable, List, Optional
 
 import requests
 
@@ -49,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:
@@ -580,7 +580,7 @@ def debug_string(self) -> str:
             buf.append(f"Env: {', '.join(envs_used)}")
         return ". ".join(buf)
 
-    def to_dict(self) -> Dict[str, any]:
+    def to_dict(self) -> Dict[str, Any]:
         return self._inner
 
     @property
@@ -713,7 +713,7 @@ def load_azure_tenant_id(self):
         self.azure_tenant_id = path_segments[1]
         logger.debug(f"Loaded tenant ID: {self.azure_tenant_id}")
 
-    def _set_inner_config(self, keyword_args: Dict[str, any]):
+    def _set_inner_config(self, keyword_args: Dict[str, Any]):
         for attr in self.attributes():
             if attr.name not in keyword_args:
                 continue

databricks/sdk/dbutils.py

@@ -188,8 +188,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
@@ -200,7 +200,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/logger/round_trip_logger.py

@@ -55,7 +55,7 @@ def generate(self) -> str:
         return "\n".join(sb)
 
     @staticmethod
-    def _mask(m: Dict[str, any]):
+    def _mask(m: Dict[str, Any]):
         for k in m:
             if k in {
                 "bytes_value",
@@ -67,7 +67,7 @@ def _mask(m: Dict[str, any]):
                 m[k] = "**REDACTED**"
 
     @staticmethod
-    def _map_keys(m: Dict[str, any]) -> List[str]:
+    def _map_keys(m: Dict[str, Any]) -> List[str]:
         keys = list(m.keys())
         keys.sort()
         return keys

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

@@ -55,7 +55,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)
@@ -198,16 +198,16 @@ class taskValues:
             def get(
                 taskKey: str,
                 key: str,
-                default: any = None,
-                debugValue: any = None,
+                default: Any = None,
+                debugValue: 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: Any) -> None:
                 """
                 Sets a task value on the current task run
                 """

docs/gen-client-docs.py

@@ -457,7 +457,7 @@ def _write_client_package_doc(self, folder: str, pkg: Package, services: list[st
 if __name__ == '__main__':
 
     @credentials_strategy('noop', [])
-    def noop_credentials(_: any):
+    def noop_credentials(_: Any):
         return lambda: {}
 
     gen = Generator()

tests/conftest.py

@@ -1,6 +1,7 @@
 import functools
 import os
 import platform
+from typing import Any
 
 import pytest as pytest
 from pyfakefs.fake_filesystem_unittest import Patcher
@@ -22,7 +23,7 @@ def stub_host_metadata(mocker):
 
 
 @credentials_strategy("noop", [])
-def noop_credentials(_: any):
+def noop_credentials(_: Any):
     return lambda: {}
 
 

tests/databricks/sdk/service/jsonmarshallv2.py

@@ -96,7 +96,7 @@ class OptionalFields:
 
     legacy_timestamp: Optional[str] = None
 
-    list_value: Optional[List[any]] = None
+    list_value: Optional[List[Any]] = None
 
     map: Optional[Dict[str, str]] = None
     """Lint disable reason: This is a dummy field used to test SDK Generation logic."""
@@ -111,13 +111,13 @@ class OptionalFields:
 
     optional_string: Optional[str] = None
 
-    struct: Optional[Dict[str, any]] = None
+    struct: Optional[Dict[str, Any]] = None
 
     test_enum: Optional[TestEnum] = None
 
     timestamp: Optional[Timestamp] = None
 
-    value: Optional[any] = None
+    value: Optional[Any] = None
 
     def as_dict(self) -> dict:
         """Serializes the OptionalFields into a dictionary suitable for use as a JSON request body."""
@@ -228,17 +228,17 @@ class RepeatedFields:
 
     repeated_int64: Optional[List[int]] = None
 
-    repeated_list_value: Optional[List[List[any]]] = None
+    repeated_list_value: Optional[List[List[Any]]] = None
 
     repeated_message: Optional[List[NestedMessage]] = None
 
     repeated_string: Optional[List[str]] = None
 
-    repeated_struct: Optional[List[Dict[str, any]]] = None
+    repeated_struct: Optional[List[Dict[str, Any]]] = None
 
     repeated_timestamp: Optional[List[Timestamp]] = None
 
-    repeated_value: Optional[List[any]] = None
+    repeated_value: Optional[List[Any]] = None
 
     test_repeated_enum: Optional[List[TestEnum]] = None
 
@@ -344,11 +344,11 @@ class RequiredFields:
 
     required_timestamp: Timestamp
 
-    required_value: any
+    required_value: Any
 
-    required_list_value: List[any]
+    required_list_value: List[Any]
 
-    required_struct: Dict[str, any]
+    required_struct: Dict[str, Any]
 
     def as_dict(self) -> dict:
         """Serializes the RequiredFields into a dictionary suitable for use as a JSON request body."""