From 2393654fe11096ba5628b4812cfde77af8c603a8 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Thu, 26 Nov 2020 10:37:43 +0100
Subject: [PATCH] bug fixed
---
utils/dead_lock_avoidance_agent.py | 2 +-
utils/fast_tree_obs.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/utils/dead_lock_avoidance_agent.py b/utils/dead_lock_avoidance_agent.py
index 37dcd0d..4a37135 100644
--- a/utils/dead_lock_avoidance_agent.py
+++ b/utils/dead_lock_avoidance_agent.py
@@ -67,7 +67,7 @@ class DeadlockAvoidanceShortestDistanceWalker(ShortestDistanceWalker):
class DeadLockAvoidanceAgent(Policy):
- def _init__(self, env: RailEnv, action_size, show_debug_plot=False):
+ def __init__(self, env: RailEnv, action_size, show_debug_plot=False):
self.env = env
self.memory = None
self.loss = 0
diff --git a/utils/fast_tree_obs.py b/utils/fast_tree_obs.py
index fa72bcf..3238ee5 100755
--- a/utils/fast_tree_obs.py
+++ b/utils/fast_tree_obs.py
@@ -36,7 +36,7 @@ class FastTreeObs(ObservationBuilder):
self.debug_render_path_list = []
if self.env is not None:
self.find_all_cell_where_agent_can_choose()
- self.dead_lock_avoidance_agent = DeadLockAvoidanceAgent(self.env, 5)
+ self.dead_lock_avoidance_agent = DeadLockAvoidanceAgent(self.env, 5, False)
else:
self.dead_lock_avoidance_agent = None
--
GitLab