search

py-why / dowhy

DoWhy is a Python library for causal inference that supports explicit modeling and testing of causal assumptions. DoWhy is based on a unified language for causal inference, combining causal graphical models and potential outcomes frameworks.
https://www.pywhy.org/dowhy
MIT License
6.99k stars 923 forks source link

Enabled float and int as possible binary treatment encoding options in class `DistanceMatchingEstimator` #987

Closed AlxndrMlk closed 1 year ago

AlxndrMlk commented 1 year ago

Enabled float and int as possible binary treatment encoding options in class DistanceMatchingEstimator.

Additionally, re-formatted some of the doc strings to match the flake8 requirements.

Other estimators discussed in https://github.com/py-why/dowhy/issues/772 already accepted int, bool, and float types and I did not change them.