From e57ff31cad442cab6b34d65d6b714b5f300ad83f Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Tue, 9 Jul 2019 11:42:26 +0200
Subject: [PATCH] #42 run baselines in ci

---
 torch_training/multi_agent_training.py | 2 +-
 torch_training/training_navigation.py  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/torch_training/multi_agent_training.py b/torch_training/multi_agent_training.py
index 216ddc6..ba9f144 100644
--- a/torch_training/multi_agent_training.py
+++ b/torch_training/multi_agent_training.py
@@ -7,7 +7,7 @@ import random
 import torch
 from dueling_double_dqn import Agent
 
-import torch_training
+import torch_training.Nets
 from flatland.envs.generators import complex_rail_generator
 from flatland.envs.observations import TreeObsForRailEnv
 from flatland.envs.predictions import ShortestPathPredictorForRailEnv
diff --git a/torch_training/training_navigation.py b/torch_training/training_navigation.py
index 095f20b..1857b67 100644
--- a/torch_training/training_navigation.py
+++ b/torch_training/training_navigation.py
@@ -8,7 +8,7 @@ import numpy as np
 import torch
 from dueling_double_dqn import Agent
 
-import torch_training
+import torch_training.Nets
 from flatland.envs.generators import complex_rail_generator
 from flatland.envs.observations import TreeObsForRailEnv
 from flatland.envs.predictions import ShortestPathPredictorForRailEnv
-- 
GitLab