Skip to content

Commit

Permalink
Added return type annotation
Browse files Browse the repository at this point in the history
  • Loading branch information
riccamini committed Jun 10, 2024
1 parent 93f30ff commit 74dbe90
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/koheesio/spark/writers/delta/batch.py
Original file line number Diff line number Diff line change
Expand Up @@ -233,7 +233,7 @@ def __merge_all(self) -> Union[DeltaMergeBuilder, DataFrameWriter]:

return self.__merge(merge_builder=builder)

def _get_merge_builder(self, provided_merge_builder=None):
def _get_merge_builder(self, provided_merge_builder=None) -> DeltaMergeBuilder:
"""Resolves the merge builder. If provided, it will be used, otherwise it will be created from the args"""

# A merge builder has been already created - case for merge_all
Expand Down

0 comments on commit 74dbe90

Please sign in to comment.