fix: changed return type of query_get_all_space_count function

dcs_core/core/common/models/validation.py

@@ -103,8 +103,6 @@ class ValidationFunction(str, Enum):
     PERCENT_EMPTY_STRING = "percent_empty_string"
     COUNT_NAN = "count_nan"
     PERCENT_NAN = "percent_nan"
-    COUNT_NULL_KEYWORD = "count_null_keyword"
-    PERCENT_NULL_KEYWORD = "percent_null_keyboard"
     COUNT_ALL_SPACE = "count_all_space"
     PERCENT_ALL_SPACE = "percent_all_space"
 

dcs_core/core/datasource/sql_datasource.py

@@ -677,7 +677,7 @@ def query_negative_metric(
 
     def query_get_all_space_count(
         self, table: str, field: str, operation: str, filters: str = None
-    ) -> int:
+    ) -> Union[int, float]:
         """
         Get the count of rows where the specified column contains only spaces.
         :param table: table name

dcs_core/core/validation/completeness_validation.py

@@ -91,24 +91,6 @@ def _generate_metric_value(self, **kwargs) -> Union[float, int]:
             raise ValueError("Invalid data source type")
 
 
-class CountEmptyStringValidation(Validation):
-    def _generate_metric_value(self, **kwargs) -> Union[float, int]:
-        if isinstance(self.data_source, SQLDataSource):
-            return self.data_source.query_get_empty_string_count(
-                table=self.dataset_name,
-                field=self.field_name,
-                filters=self.where_filter if self.where_filter is not None else None,
-            )
-        elif isinstance(self.data_source, SearchIndexDataSource):
-            return self.data_source.query_get_empty_string_count(
-                index_name=self.dataset_name,
-                field=self.field_name,
-                filters=self.where_filter if self.where_filter else None,
-            )
-        else:
-            raise ValueError("Invalid data source type")
-
-
 class CountAllSpaceValidation(Validation):
     def _generate_metric_value(self, **kwargs) -> Union[float, int]:
         if isinstance(self.data_source, SQLDataSource):