Consider this import:
from .scale_shape import scale_shape
https://github.com/has2k1/plotnine/blob/cbc50afaccb82e6f062034139cfc190fc49f81c5/plotn