From b84e195292625a131cc0485dc539ec1c8a8abcfa Mon Sep 17 00:00:00 2001 From: Matt Delacour Date: Tue, 15 Nov 2022 10:48:30 -0500 Subject: [PATCH] Update typing of TableSegment().count() On my side when dst is a TableSegment object, dst.count() returns an int. Using `data-diff==0.2.8` --- data_diff/table_segment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data_diff/table_segment.py b/data_diff/table_segment.py index cddbe9f5..a9544a12 100644 --- a/data_diff/table_segment.py +++ b/data_diff/table_segment.py @@ -154,7 +154,7 @@ def relevant_columns(self) -> List[str]: def _relevant_columns_repr(self) -> List[Expr]: return [NormalizeAsString(this[c]) for c in self.relevant_columns] - def count(self) -> Tuple[int, int]: + def count(self) -> int: """Count how many rows are in the segment, in one pass.""" return self.database.query(self.make_select().select(Count()), int)