atari_gameplay_for_chromosome.py

def _gameplay_for_chromosome(self, chromosome, env):
        self.run += 1
        self.logger.add_run(self.run)

        self.model.set_weights(chromosome)
        state = env.reset()
        score = 0
        while True:
            action = self._predict(state)
            state_next, reward, terminal, info = env.step(action)
            score += np.sign(reward)
            state = state_next
            if terminal:
                self.logger.add_score(score)
                return score