From 462c6e314b9fcafcec3095d7bc283b223a2b437d Mon Sep 17 00:00:00 2001
From: itaicaspi-intel <itai.caspi@intel.com>
Date: Thu, 24 May 2018 15:15:33 +0300
Subject: [PATCH] bug fix in nec checkpoint saving

---
 agents/nec_agent.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/agents/nec_agent.py b/agents/nec_agent.py
index 77520a4..a327db4 100644
--- a/agents/nec_agent.py
+++ b/agents/nec_agent.py
@@ -15,7 +15,7 @@
 #
 
 import numpy as np
-
+import os, pickle
 from agents.value_optimization_agent import ValueOptimizationAgent
 from logger import screen
 from utils import RunPhase