fix integration tests

sdv-dev · Feb 16, 2022 · b3c1f3f · b3c1f3f
1 parent d587742
commit b3c1f3f
Show file tree

Hide file tree

Showing 7 changed files with 426 additions and 459 deletions.
diff --git a/sdv/tabular/base.py b/sdv/tabular/base.py
@@ -372,7 +372,7 @@ def _make_condition_dfs(self, conditions):
         ]
 
     def _conditionally_sample_rows(self, dataframe, condition, transformed_condition,
-                                   max_tries=None, batch_size_per_try=None, float_rtol=None,
+                                   max_tries=None, batch_size_per_try=None, float_rtol=0.01,
                                    graceful_reject_sampling=True):
         num_rows = len(dataframe)
         sampled_rows = self._sample_batch(