Index: scheduler/drone_manager.py |
diff --git a/scheduler/drone_manager.py b/scheduler/drone_manager.py |
index e094f14f1fb080ef9108b90b287b0b46f49e78ae..75724f382b83c0849b01ff2ed5194134ab4c0ff1 100644 |
--- a/scheduler/drone_manager.py |
+++ b/scheduler/drone_manager.py |
@@ -159,7 +159,8 @@ class DroneManager(object): |
self._results_dir = base_results_dir |
for hostname in drone_hostnames: |
- self._add_drone(hostname) |
+ drone = self._add_drone(hostname) |
+ drone.call('initialize', self.absolute_path('')) |
if not self._drones: |
# all drones failed to initialize |
@@ -204,9 +205,8 @@ class DroneManager(object): |
def _add_drone(self, hostname): |
logging.info('Adding drone %s' % hostname) |
drone = drones.get_drone(hostname) |
- if drone: |
- self._drones[drone.hostname] = drone |
- drone.call('initialize', self.absolute_path('')) |
+ self._drones[drone.hostname] = drone |
+ return drone |
def _remove_drone(self, hostname): |