def param_shapes()

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

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

    def param_shapes(self, shape: torch.Size) -> Sequence[torch.Size]:
        s1 = torch.Size(shape + (self.count_bins,))
        s2 = torch.Size(shape + (self.count_bins - 1,))

        if self.order == "linear":
            return s1, s1, s2, s1
        else:
            return s1, s1, s2