def setup()

in flows.py [0:0]

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

    def setup(self):
        if self.cost_gamma == 'None': self.cost_gamma = None
        if self.min_zero_gamma == 'None': self.min_zero_gamma = None

        if isinstance(self.min_zero_gamma, str):
            self.min_zero_gamma = float(self.min_zero_gamma)

        if isinstance(self.manifold, Product):
            mu_init = densities.get(self.manifold, 'ProductUniformComponents')
        else:
            mu_init = densities.get_uniform(self.manifold)

        self.mus = self.param(
            'mus', init_manifold_samples(mu_init),
            [self.manifold.D, self.n_components])
        if self.init_alpha_mode == 'linear':
            alphas = self.init_alpha_linear_scale*self.mus[:,0].dot(self.mus)
            self.alphas = self.param(
                'alphas', lambda key, shape: alphas,
                [self.n_components])
        elif self.init_alpha_mode == 'uniform':
            self.alphas = self.param(
                'alphas', init_uniform(
                    minval=self.init_alpha_minval,
                    maxval=self.init_alpha_minval+self.init_alpha_range),
                [self.n_components])
        else:
            assert False