def __init__()

in latent_operators.py [0:0]

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

    def __init__(self, cardinals, z_dim, device, unique_transfo=False, index=None):
        self.cardinals = cardinals
        self.z_dim = z_dim
        self.device = device
        self.translation_matrices = self.generate_translation_matrices(
            self.cardinals, self.z_dim
        )
        if unique_transfo:
            if (np.array(cardinals)>1).sum()==1:
                self.index = int((np.array(cardinals)>1).nonzero()[0])
            elif (np.array(cardinals)>1).sum()>1:
                if index is None:
                    print("Must provide the index of the operator !")
                else:
                    self.index = index
            self.translate_batch = self.translate_batch_unique
        else:
            self.translate_batch = self.translate_batch_multiple