5ab90a008e1e5288ccdae067a38e00a044608948,General_Deep_Q_RL/agent.py,NeuralAgent,run,#NeuralAgent#,123

Before Change

    def run(self, nEpochs, epochLength):
        for c in self._controllers: c.OnStart(self)
        for _ in range(nEpochs):
            if self._mode != -1:
                while self._modeEpochsLength > 0:
                    self._modeEpochsLength = self._runEpisode(self._modeEpochsLength)
            else:
                length = epochLength
                while length > 0:
                    length = self._runEpisode(length)

            for c in self._controllers: c.OnEpochEnd(self)
            
        for c in self._controllers: c.OnEnd(self)

    def _runEpisode(self, maxSteps):
        self._inEpisode = True

After Change

    def run(self, nEpochs, epochLength):
        for c in self._controllers: c.OnStart(self)
        i = 0
        while i < nEpochs:
            if self._mode != -1:
                while self._modeEpochsLength > 0:
                    self._modeEpochsLength = self._runEpisode(self._modeEpochsLength)
            else:
                length = epochLength
                while length > 0:
                    length = self._runEpisode(length)
                i += 1

            for c in self._controllers: c.OnEpochEnd(self)
            
        for c in self._controllers: c.OnEnd(self)

    def _runEpisode(self, maxSteps):
        self._inEpisode = True

Instances