From 0b0288895e47456be74f1fb382306b90be1714f1 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Sat, 31 Aug 2019 21:29:03 -0400
Subject: [PATCH] updated loading of files in example files
---
torch_training/multi_agent_inference.py | 2 ++
torch_training/observation_builders/__init__.py | 0
torch_training/predictors/__init__.py | 0
3 files changed, 2 insertions(+)
delete mode 100644 torch_training/observation_builders/__init__.py
delete mode 100644 torch_training/predictors/__init__.py
diff --git a/torch_training/multi_agent_inference.py b/torch_training/multi_agent_inference.py
index 66a37ad..3fc6468 100644
--- a/torch_training/multi_agent_inference.py
+++ b/torch_training/multi_agent_inference.py
@@ -10,6 +10,7 @@ from predictors.predictions import ShortestPathPredictorForRailEnv
import torch_training.Nets
from flatland.envs.rail_env import RailEnv
from flatland.envs.rail_generators import rail_from_file
+from flatland.envs.schedule_generators import schedule_from_file
from flatland.utils.rendertools import RenderTool
from torch_training.dueling_double_dqn import Agent
from utils.observation_utils import normalize_observation
@@ -24,6 +25,7 @@ file_name = "./railway/simple_avoid.pkl"
env = RailEnv(width=10,
height=20,
rail_generator=rail_from_file(file_name),
+ schedule_generator=schedule_from_file(file_name),
obs_builder_object=observation_helper)
x_dim = env.width
y_dim = env.height
diff --git a/torch_training/observation_builders/__init__.py b/torch_training/observation_builders/__init__.py
deleted file mode 100644
index e69de29..0000000
diff --git a/torch_training/predictors/__init__.py b/torch_training/predictors/__init__.py
deleted file mode 100644
index e69de29..0000000
--
GitLab