From a8fabfb47a35130e24be468236ca7b230349d067 Mon Sep 17 00:00:00 2001
From: Mohamed Zahran <moh3th1@gmail.com>
Date: Thu, 13 Dec 2018 01:46:16 +0200
Subject: [PATCH] Update envs to be compatible with python 3.X

---
 envs.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/envs.py b/envs.py
index 1dc2d91..f8592bb 100644
--- a/envs.py
+++ b/envs.py
@@ -87,7 +87,7 @@ def _get_val(self):
 
   def _trade(self, action):
     # all combo to sell(0), hold(1), or buy(2) stocks
-    action_combo = map(list, itertools.product([0, 1, 2], repeat=self.n_stock))
+    action_combo = list(map(list, itertools.product([0, 1, 2], repeat=self.n_stock)))
     action_vec = action_combo[action]
 
     # one pass to get sell/buy index