From 4e1d484b63b6c48059d1b11a00ff3f3b99295457 Mon Sep 17 00:00:00 2001
From: Pavel Lutskov <pavel.lutskov@gmail.com>
Date: Sun, 2 Aug 2020 12:21:20 +0200
Subject: [PATCH] I guess formulate the G2 cleaner

---
 main.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index ffd3381..77dcb23 100644
--- a/main.py
+++ b/main.py
@@ -67,9 +67,8 @@ def init_global(maze_filename):
     maze_cost[MAZE == 'T'] = 50
     maze_cost[MAZE == 'G'] = -1
     G1_X = maze_cost.copy()[state_mask]
-    # maze_cost[(MAZE=='0') | (MAZE=='S') | (MAZE=='G')] += 1
-    # G2_X = maze_cost.copy()[state_mask]
-    G2_X = G1_X + 1
+    maze_cost[(MAZE=='0') | (MAZE=='S') | (MAZE=='G')] += 1
+    G2_X = maze_cost.copy()[state_mask]
 
     # Actual environment modelling
     U_OF_X = np.zeros((SN, len(A2)), dtype=np.bool)