def setup()

in flows.py [0:0]

• 14 lines of code
• 1 McCabe index (conditional complexity)

    def setup(self):
        assert isinstance(self.manifold, Sphere)
        mu_init = densities.get_uniform(self.manifold)
        self.betas = self.param(
            'betas', init_uniform(
                minval=self.init_beta_minval,
                maxval=self.init_beta_minval+self.init_beta_range
            ), [self.n_radial_components])
        self.mus = self.param(
            'mus', init_manifold_samples(mu_init),
            [self.manifold.D, self.n_radial_components])
        self.alphas = self.param(
            'alphas', init_full(1./self.n_radial_components),
            [self.n_radial_components])