diff --git a/pyfixest/experimental/did.py b/pyfixest/experimental/did.py index cbc9bdc1..1f17fa58 100644 --- a/pyfixest/experimental/did.py +++ b/pyfixest/experimental/did.py @@ -14,6 +14,7 @@ + def event_study(data, yname, idname, tname, gname, xfml = None, estimator = "twfe", att = True, cluster = "idname"): """ @@ -337,6 +338,7 @@ def _did2s_estimate(data: pd.DataFrame, yname : str, _first_stage: str, _second_ # estimate first stage fit1 = feols(fml = _first_stage_full, data = _not_yet_treated_data, vcov = "iid") # iid as it might be faster than CRV + # obtain estimated fixed effects fit1.fixef() diff --git a/readme.md b/readme.md index 99db40bf..e75e2f8e 100644 --- a/readme.md +++ b/readme.md @@ -17,7 +17,7 @@ At the moment, `PyFixest` supports - Several Robust and Cluster Robust Variance-Covariance Types - Wild Cluster Bootstrap Inference (via [wildboottest](https://github.com/s3alfisc/wildboottest)) - Support for estimators of the "new" Difference-in-Difference literature is work in progress. `PyFixest` currently provides an - experimentla implementation of Gardner's Did2s estimtator (via the `pyfixest.experimental.did` module, only ATT estimation). + experimental implementation of Gardner's Did2s estimtator (via the `pyfixest.experimental.did` module, only ATT estimation). ## Installation diff --git a/tests/test_did.py b/tests/test_did.py index ccbb23b6..48b5d14b 100644 --- a/tests/test_did.py +++ b/tests/test_did.py @@ -20,6 +20,7 @@ def test_event_study(): Test the event_study() function. """ + df_het = pd.read_csv("pyfixest/experimental/data/df_het.csv") fit_did2s = event_study( @@ -86,4 +87,3 @@ def test_did2s(): ) -