Update __init__.py

[seduerr91]

With the change in the python 3 release 7 the yield function started to work differently, hence this adaption is made. Thank you