2cb8599c0ea174253d05c390c1eb7e2eed1d88ec,app/grandchallenge/datasets/models.py,IndexMixin,index,#IndexMixin#,24

Before Change

    def index(self: Union["ImageSet", "AnnotationSet"]):
        images = self.images.all()
        common_prefix = commonprefix([i.name.lower() for i in images])
        return {i.sorter_key(start=len(common_prefix)): i for i in images}


class ImageSet(UUIDModel, IndexMixin):
    TRAINING = "TRN"

After Change

        images = self.images.all()
        common_prefix = commonprefix([i.name.lower() for i in images])
        if include_shape:
            return {
                (
                    find_first_int(instr=i.name[len(common_prefix) :]),
                    *i.shape,
                ): i
                for i in images
            }
        else:
            return {
                find_first_int(instr=i.name[len(common_prefix) :]): i
                for i in images

Instances