Change total row number computation

src/whylogs/core/datasetprofile.py

@@ -123,8 +123,6 @@ def __init__(
 
         self.model_profile = model_profile
 
-        self.column_row_dict = dict()
-
         # Store Name attribute
         self._tags["name"] = name
 
@@ -167,8 +165,7 @@ def session_timestamp_ms(self):
 
     @property
     def total_row_number(self):
-        dict_counts = self.column_row_dict.values() if len(self.column_row_dict) else [0]
-        return max(dict_counts)
+        return max([col_prof.counters.count for col_prof in self.columns.values()])
 
     def add_output_field(self, field: Union[str, List[str]]):
         if self.model_profile is None:
@@ -265,11 +262,6 @@ def track_datum(self, column_name, data, character_list=None, token_method=None)
             prof = ColumnProfile(column_name, constraints=constraints)
             self.columns[column_name] = prof
 
-            self.column_row_dict[column_name] = 0
-
-        # updating the map for every column name with increasing the number of tracked values
-        self.column_row_dict[column_name] += 1
-
         prof.track(data, character_list=None, token_method=None)
 
     def track_multi_column(self, columns):