fix bug related to hasattr

Author: waffoo

accel/agents/sac.py

@@ -251,7 +251,7 @@ def _eval_one_episode(self, env, frames, record=False,
                     self._add_obs_to_frame(img, frames)
 
                 total_reward += reward
-                if hasattr(info, 'success'):
+                if 'success' in info.keys():
                     success |= bool(info['success'])
 
             if not hasattr(env, 'was_real_done') or env.was_real_done:

examples/atari_dqn.py

@@ -170,8 +170,7 @@ def main(cfg):
                 step += 1
 
                 timeout_label = 'TimeLimit.truncated'
-                timeout = hasattr(
-                    info, timeout_label) and info[timeout_label]
+                timeout = timeout_label in info.keys() and info[timeout_label]
                 next_valid = 1. if timeout else float(not done)
                 agent.update(obs, action, next_obs, reward, next_valid)