chore: fix linting

Author: Caio Fontes

elasticsearch_dsl/function.py

@@ -17,12 +17,27 @@
 
 import collections.abc
 from copy import deepcopy
-from typing import Dict, Optional, ClassVar, Union, MutableMapping, Any
+from typing import Any, ClassVar, Dict, MutableMapping, Optional, Union, overload
 
 from .utils import DslBase, _JSONSafeTypes
 
 
-def SF(name_or_sf: Union[str, "ScoreFunction", MutableMapping[str, Any]], **params: Any) -> "ScoreFunction":
+@overload
+def SF(name_or_sf: MutableMapping[str, Any]) -> "ScoreFunction": ...
+
+
+@overload
+def SF(name_or_sf: "ScoreFunction") -> "ScoreFunction": ...
+
+
+@overload
+def SF(name_or_sf: str, **params: Any) -> "ScoreFunction": ...
+
+
+def SF(
+    name_or_sf: Union[str, "ScoreFunction", MutableMapping[str, Any]],
+    **params: Any,
+) -> "ScoreFunction":
     # {"script_score": {"script": "_score"}, "filter": {}}
     if isinstance(name_or_sf, collections.abc.MutableMapping):
         if params:

elasticsearch_dsl/utils.py

@@ -18,13 +18,15 @@
 
 import collections.abc
 from copy import copy
-from typing import Any, Dict, Optional, Type, ClassVar, Union, List
+from typing import Any, ClassVar, Dict, List, Optional, Type, Union
 
 from typing_extensions import Self
 
 from .exceptions import UnknownDslObject, ValidationException
 
-_JSONSafeTypes = Union[int, bool, str, float, List["_JSONSafeTypes"], Dict[str, "_JSONSafeTypes"]]
+_JSONSafeTypes = Union[
+    int, bool, str, float, List["_JSONSafeTypes"], Dict[str, "_JSONSafeTypes"]
+]
 
 SKIP_VALUES = ("", None)
 EXPAND__TO_DOT = True
@@ -212,7 +214,7 @@ class DslMeta(type):
     For typical use see `QueryMeta` and `Query` in `elasticsearch_dsl.query`.
     """
 
-    _types: ClassVar[Dict[str, type["DslBase"]]] = {}
+    _types: ClassVar[Dict[str, Type["DslBase"]]] = {}
 
     def __init__(cls, name, bases, attrs):
         super().__init__(name, bases, attrs)