Convert to timedelta

sdv-dev · Jun 29, 2023 · 9692450 · 9692450
1 parent bff5913
commit 9692450
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdv/constraints/tabular.py b/sdv/constraints/tabular.py
@@ -712,7 +712,7 @@ def _reverse_transform(self, table_data):
             diff_column = diff_column.round()
 
         if self._is_datetime:
-            diff_column = diff_column.astype('timedelta64[ns]', errors='ignore')
+            diff_column = convert_to_timedelta(diff_column)
 
         if self._operator in [np.greater, np.greater_equal]:
             original_column = self._value + diff_column