def get_parent_from_bijector()

in flowtorch/bijectors/bijective_tensor.py [0:0]

• 9 lines of code
• 5 McCabe index (conditional complexity)

    def get_parent_from_bijector(self, bijector: "Bijector") -> Tensor:
        if self._bijector is bijector:
            return self.parent
        for parent in self.parents():
            if not isinstance(parent, BijectiveTensor):
                break
            if parent._bijector is bijector:
                return parent.parent
        raise RuntimeError("bijector not found in flow")