mypy: ensure default are set in overloaded methods (#698)

quixio · Jan 6, 2025 · c03a9e8 · c03a9e8
1 parent 7d67658
commit c03a9e8
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/quixstreams/core/stream/stream.py b/quixstreams/core/stream/stream.py
@@ -113,7 +113,7 @@ def __repr__(self) -> str:
         return f"<{self.__class__.__name__} [{len(tree_funcs)}]: {funcs_repr}>"
 
     @overload
-    def add_filter(self, func: FilterCallback, *, metadata: Literal[False]):
+    def add_filter(self, func: FilterCallback, *, metadata: Literal[False] = False):
         pass
 
     @overload
@@ -149,7 +149,11 @@ def add_filter(
 
     @overload
     def add_apply(
-        self, func: ApplyCallback, *, expand: Literal[False], metadata: Literal[False]
+        self,
+        func: ApplyCallback,
+        *,
+        expand: Literal[False] = False,
+        metadata: Literal[False] = False,
     ):
         pass
 
@@ -159,7 +163,7 @@ def add_apply(
         func: ApplyExpandedCallback,
         *,
         expand: Literal[True],
-        metadata: Literal[False],
+        metadata: Literal[False] = False,
     ):
         pass
 
@@ -168,7 +172,7 @@ def add_apply(
         self,
         func: ApplyWithMetadataCallback,
         *,
-        expand: Literal[False],
+        expand: Literal[False] = False,
         metadata: Literal[True],
     ):
         pass
@@ -217,7 +221,7 @@ def add_apply(
         return self._add(apply_func)
 
     @overload
-    def add_update(self, func: UpdateCallback, *, metadata: Literal[False]):
+    def add_update(self, func: UpdateCallback, *, metadata: Literal[False] = False):
         pass
 
     @overload
@@ -251,7 +255,7 @@ def add_update(
         return self._add(update_func)
 
     @overload
-    def add_transform(self, func: TransformCallback, *, expand: Literal[False]):
+    def add_transform(self, func: TransformCallback, *, expand: Literal[False] = False):
         pass
 
     @overload