Fixing last minute mistakes

carla-simulator · Apr 23, 2018 · 121944b · 121944b
1 parent 1c78c1d
commit 121944b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/agents/imitation/imitation_learning.py b/agents/imitation/imitation_learning.py
@@ -7,7 +7,7 @@
 
 slim = tf.contrib.slim
 
-from carla.benchmarks.agent import Agent
+from carla.agent import Agent
 from carla.carla_server_pb2 import Control
 from agents.imitation.imitation_learning_network import load_imitation_learning_network
 
@@ -16,7 +16,7 @@ class ImitationLearning(Agent):
 
     def __init__(self, city_name, avoid_stopping, memory_fraction=0.25, image_cut=[115, 510]):
 
-        Agent.__init__(self, city_name)
+        Agent.__init__(self)
 
         self.dropout_vec = [1.0] * 8 + [0.7] * 2 + [0.5] * 2 + [0.5] * 1 + [0.5, 1.] * 5
 

diff --git a/run_CIL.py b/run_CIL.py
@@ -2,7 +2,7 @@
 import logging
 
 from carla.driving_benchmark import run_driving_benchmark
-from carla.driving_benchmark.experiment_suite import CoRL2017
+from carla.driving_benchmark.experiment_suites import CoRL2017
 
 from agents.imitation.imitation_learning import ImitationLearning