def correct_channel()

in src_batch_job/source/model.py [0:0]

• 6 lines of code
• 2 McCabe index (conditional complexity)

    def correct_channel(self, image):
        if (len(image.shape) == 2):
            # Correct one channel (black-write) image to three channel (RGB) image by stacking
            image = nd.stack(image, image, image, axis=2)

        assert len(image.shape) == 3
        assert image.shape[2] == 3
        return image