def forward()

in src/image_gen_aux/preprocessors/teed/teed.py [0:0]

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

    def forward(self, x):
        x = self.conv1(x)
        x = self.smish(x)
        x = self.conv2(x)
        if self.use_act:
            x = self.smish(x)
        return x