def downscale(self, images, scale): if self._scale_mode == 'ALL': return layers.downscale(images, scale) elif self._scale_mode == 'H': return layers.downscale_height(images, scale) else: self._raise_unsupported_scale_mode_error()