From 04a942e5eec345d9d3c64a454e59f331abd7a278 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Sat, 28 Nov 2020 20:26:49 +0100
Subject: [PATCH] bug fixed

---
 run.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/run.py b/run.py
index 55e31ea..33bc56c 100644
--- a/run.py
+++ b/run.py
@@ -45,7 +45,7 @@ from reinforcement_learning.dddqn_policy import DDDQNPolicy
 
 # Print per-step logs
 VERBOSE = True
-USE_FAST_TREE_OBS = True
+USE_FAST_TREEOBS = True
 
 # Checkpoint to use (remember to push it!)
 checkpoint = "./checkpoints/201124171810-7800.pth"  # 18.249244799876152 DEPTH=2 AGENTS=10
@@ -74,9 +74,11 @@ if USE_FAST_TREEOBS:
     def check_is_observation_valid(observation):
         return True
 
+
     def get_normalized_observation(observation, tree_depth: int, observation_radius=0):
         return observation
 
+
     tree_observation = FastTreeObs(max_depth=observation_tree_depth)
     state_size = tree_observation.observation_dim
 else:
-- 
GitLab