def __init__(self, *args, **kwargs): super(TestBaseAgent, self).__init__(*args, **kwargs) self.N = np.random.randint(1, 100) self.s = BaseSupervisor(self.N) self.a = BaseAgent(0, self.s)