Skip to content

Commit

Permalink
chore: fix pandas flaky test
Browse files Browse the repository at this point in the history
  • Loading branch information
RomanBredehoft committed Jun 7, 2024
1 parent f595c46 commit 948cd5b
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tests/pandas/test_pandas.py
Original file line number Diff line number Diff line change
Expand Up @@ -714,7 +714,9 @@ def check_invalid_schema_values():
):
client.encrypt_from_pandas(pandas_df, schema=schema_float_column)

schema_float_oob = {f"{feat_name}_float_1": {"min": float_min // 2, "max": float_max // 2}}
# Here, we want to make sure that either min or max is out of the distribution. We make sure
# this is always the case (not subject to randomization) by setting 'min' to 'float_max'
schema_float_oob = {f"{feat_name}_float_1": {"min": float_max, "max": float_max}}

with pytest.raises(
ValueError,
Expand Down

0 comments on commit 948cd5b

Please sign in to comment.