def set_transform()

in src/lic/ppl/world/variable.py [0:0]

• 18 lines of code
• 8 McCabe index (conditional complexity)

    def set_transform(self, transform_data: TransformData, proposer):
        """
        Sets the variable transform to the transform passed in.

        :param transform: the transform value to the set the variable transform
        to
        """
        if transform_data.transform_type == TransformType.DEFAULT:
            self.transforms = get_default_transforms(self.distribution)
        elif transform_data.transform_type == TransformType.NONE:
            if (
                isinstance(self.distribution, dist.Beta)
                and proposer is not None
                and hasattr(proposer, "reshape_untransformed_beta_to_dirichlet")
                and proposer.reshape_untransformed_beta_to_dirichlet
            ):
                self.transforms = [BetaDimensionTransform()]
            else:
                self.transforms = []
        else:
            if transform_data.transforms is None:
                self.transforms = []
            else:
                # pyre-fixme
                self.transforms = transform_data.transforms