more

.codegen/service.py.tmpl

@@ -3,7 +3,7 @@
 from dataclasses import dataclass
 from datetime import timedelta
 from enum import Enum
-from typing import Dict, List, Any, Optional, Iterator, Type, Callable
+from typing import Dict, List, Any, Optional, Iterator, Type, Callable, Mapping
 import time
 import random
 import logging

databricks/sdk/service/_internal.py

@@ -1,5 +1,5 @@
 import datetime
-from typing import Any, Callable, Dict, Generic, Optional, Type, TypeVar
+from typing import Any, Callable, Dict, Generic, Optional, Type, TypeVar, Mapping
 
 
 def _from_dict(d: Dict[str, Any], field: str, cls: Type) -> Any:
@@ -29,7 +29,7 @@ class Wait(Generic[ReturnType]):
     def __init__(self,
                  waiter: Callable,
                  response: Any = None,
-                 **kwargs) -> None: # type: ignore[no-untyped-def]
+                 **kwargs: Mapping[str, Any]) -> None:
         self.response = response
 
         self._waiter = waiter