Error: TypeError: __init__() takes from 1 to 3 positional arguments but 5 were given

Question

I was translating a program from pytorch to Tensorflow in which I wrote for a generator model:

class Generator(tf.Module):
    def __init__(self, noise_size, cond