From 8947c15031310084968d402828b93d79f1a4a60b Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Thu, 4 Jul 2019 16:41:12 -0400
Subject: [PATCH] bugfix in observation and prediction for tree observation

---
 examples/training_example.py | 2 --
 notebooks/Scene_Editor.ipynb | 2 +-
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/examples/training_example.py b/examples/training_example.py
index cdbd1ad..dd9ded9 100644
--- a/examples/training_example.py
+++ b/examples/training_example.py
@@ -76,8 +76,6 @@ for trials in range(1, n_trials + 1):
         for a in range(env.get_num_agents()):
             action = agent.act(obs[a])
             action_dict.update({a: action})
-            # Uncomment next line to print observation of an agent
-            # TreeObservation.util_print_obs_subtree((obs[a]))
 
         # Environment step which returns the observations for all agents, their corresponding
         # reward and whether their are done
diff --git a/notebooks/Scene_Editor.ipynb b/notebooks/Scene_Editor.ipynb
index a36b4e1..64bbad7 100644
--- a/notebooks/Scene_Editor.ipynb
+++ b/notebooks/Scene_Editor.ipynb
@@ -70,7 +70,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "4153480c3fbe4fd9864a4d52b416477c",
+       "model_id": "0b92e7084e37450cbfb3855dc3a58543",
        "version_major": 2,
        "version_minor": 0
       },
-- 
GitLab