fix test

sdv-dev · Feb 18, 2022 · 1098a0d · 1098a0d
1 parent dcfc5e5
commit 1098a0d
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/integration/tabular/test_base.py b/tests/integration/tabular/test_base.py
@@ -265,7 +265,6 @@ def test_conditional_sampling_constraint_uses_reject_sampling(gm_mock):
     })
     sample_calls = model._model.sample.mock_calls
     assert len(sample_calls) == 2
-    model._model.sample.assert_any_call(5, conditions=expected_transformed_conditions)
     model._model.sample.assert_any_call(50, conditions=expected_transformed_conditions)
     pd.testing.assert_frame_equal(sampled_data, expected_data)