I don't think this class exists anymore on `master`! Now that `fit()` works properly we can just use `centerCrop()` instead of this custom stuff.