def param_shapes()

in flowtorch/bijectors/ops/affine.py [0:0]

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

    def param_shapes(self, shape: torch.Size) -> Tuple[torch.Size, torch.Size]:
        # A mean and log variance for every dimension of the event shape
        return shape, shape