diff --git a/pl_bolts/models/gans/pix2pix/components.py b/pl_bolts/models/gans/pix2pix/components.py index a686dd4482..ee85c60645 100644 --- a/pl_bolts/models/gans/pix2pix/components.py +++ b/pl_bolts/models/gans/pix2pix/components.py @@ -36,7 +36,7 @@ def __init__( kernel: int = 4, strides: int = 2, padding: int = 1, - batchnorm: bool = True + batchnorm: bool = True, ) -> None: super().__init__() layers = [nn.Conv2d(in_channels, out_channels, kernel, strides, padding)]