MAINT: special: fix import in generate_ufuncs.py