I think this should be `data : None` and add a couple of lines to the docstring explaining that base classes can override this method to return a tuple of image-shaped data and custom data.