polsys / ennemi

Easy Nearest Neighbor Estimation of Mutual Information
https://polsys.github.io/ennemi/
MIT License
35 stars 12 forks source link

Properly annotate DataFrame returns #92

Open polsys opened 3 years ago

polsys commented 3 years ago

estimate_mi and pairwise_mi return DataFrames when passed ones. Investigate how to annotate this case. Note that this might need a hard dependency on pandas. (Related to #48.)

This would let us remove some manual annotations in pandas tests, and any annotated user code (which I'm skeptical about).

Originally posted by @polsys in https://github.com/polsys/ennemi/issues/91#issuecomment-892488466

polsys commented 7 months ago

Maybe related to #121