Revert "python iml change"

This reverts commit eee5838eb48658b024211721f1c98a1849dc82e8.