From 7725dabc8608509554110acae3b2448867343398 Mon Sep 17 00:00:00 2001 From: itaicaspi-intel Date: Sat, 26 May 2018 17:49:13 +0300 Subject: [PATCH] checkpoints bug fix --- agents/agent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/agents/agent.py b/agents/agent.py index 0ab4db3..010d5d8 100644 --- a/agents/agent.py +++ b/agents/agent.py @@ -550,7 +550,7 @@ class Agent(object): if current_snapshot_period > model_snapshots_periods_passed: model_snapshots_periods_passed = current_snapshot_period self.save_model(model_snapshots_periods_passed) - if self.running_observation_stats is not None: + if hasattr(self, 'running_observation_state') and self.running_observation_stats is not None: to_pickle(self.running_observation_stats, os.path.join(self.tp.save_model_dir, "running_stats.p".format(model_snapshots_periods_passed)))