ENH: add PyPy compatibility