ColumnFormula raises ConstraintsNotMetError when checking NaN values

[npatki]

Environment Details

• SDV version: 0.12.0

Error Description

I have written a ColumnFormula constraint that I know is correct. (I have used it on the real data and confirmed it outputs the same values.)

However, whenever I apply it to my model, I get an error: ConstraintsNotMetError: Data is not valid for the given constraints. Upon closer inspection, I think it's because:

• My ColumnFormula will sometimes output NaN values (for eg. [1, 2, NaN, 2, 3, NaN, NaN]) -- same values exist in my real data
• The SDV is using a simple == to check equivalence (ref). Since NaN != NaN, it thinks my formula's output is invalid.

Expected Outcome

I expect the ColumnFormula should consider NaN == NaN for the purposes of validity.
Can we use some sort of other check here to ensure that the formula passes when it includes NaNs?