Override EA function in a python class