fixed warning (#464)

This commit is contained in:
Clayton Thorrez
2018-09-05 18:12:01 -04:00
committed by pzhokhov
parent 43ed76944b
commit 1e9051e87e

View File

@@ -116,7 +116,7 @@ class RolloutWorker:
return self.generate_rollouts() return self.generate_rollouts()
if np.isnan(o_new).any(): if np.isnan(o_new).any():
self.logger.warning('NaN caught during rollout generation. Trying again...') self.logger.warn('NaN caught during rollout generation. Trying again...')
self.reset_all_rollouts() self.reset_all_rollouts()
return self.generate_rollouts() return self.generate_rollouts()