Better handling of treated input in RegressionDiscontinuity

[drbenvincent]

When doing regression discontinuity analysis, eg.

result = cp.RegressionDiscontinuity(
    df,
    formula="y ~ 1 + x + treated + x:treated",
    model=cp.pymc_models.LinearRegression(sample_kwargs={"random_seed": seed}),
    treatment_threshold=0.5,
)

it looks like treated has to be of type bool. A mysterious error arises if it is instead 0's and 1's coded as int's.

• Add an extra data validation step
• Add a test to check that we get an exception if we provide ints

[inhandan]

I'd like to solve this. Can you provide a code snippet and error message? Please include definition of the df, and in particular the treated column