david-cortes / contextualbandits

Python implementations of contextual bandits algorithms
http://contextual-bandits.readthedocs.io
BSD 2-Clause "Simplified" License
739 stars 143 forks source link

remove an assertion for the reward estimator #37

Closed YanjieHe closed 3 years ago

YanjieHe commented 3 years ago

The variable X was not defined in the scope. How about removing or commenting the assertion?

assert reward_estimator.shape[0] == X.shape[0]
david-cortes commented 3 years ago

Thanks, merged and pushed to pypi.