def get_directional_transform_name()

in similarity_search_experiments/correlate_rank_with_invariance_gap.py [0:0]

• 14 lines of code
• 8 McCabe index (conditional complexity)

    def get_directional_transform_name(self, rank_name: str) -> str:
        is_exception = self._is_exception(rank_name)
        transform = rank_name.split(" ")[0]
        if not is_exception:
            return transform

        magnitude = int(rank_name.split(" ")[-1])

        if transform == "rescale" and magnitude <= 5:
            return "scale_zoom_in"
        elif transform == "rescale":
            return "scale_zoom_in"
        elif transform == "contrast" and magnitude <= 5:
            return "decrease_contrast"
        elif transform == "contrast":
            return "increase_contrast"