def set_available_actions(self, actions): super(SLAgent, self).set_available_actions(actions) # possible state values state_n = len(self.preprocessor.enumerate_states()) self.nn = MLP(config='simple', input_ranges=[[0, state_n]], n_outputs=len(actions), batch_size=4)