empty dataframy bug fix

sdv-dev · Jan 7, 2025 · 5754179 · 5754179
1 parent f63ec91
commit 5754179
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/sdv/data_processing/data_processor.py b/sdv/data_processing/data_processor.py
@@ -965,6 +965,8 @@ def filter_valid(self, data):
                 Table containing only the valid rows.
         """
         for constraint in self._constraints:
+            if data.empty:
+                return data
             data = constraint.filter_valid(data)
 
         return data

diff --git a/sdv/single_table/base.py b/sdv/single_table/base.py
@@ -637,6 +637,8 @@ def _filter_conditions(sampled, conditions, float_rtol):
             pandas.DataFrame:
                 Rows from the sampled data that match the conditions.
         """
+        if sampled.empty:
+            return sampled
         for column, value in conditions.items():
             column_values = sampled[column]
             if column_values.dtype.kind == 'f':