aboutsummaryrefslogtreecommitdiff
path: root/rumba/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'rumba/utils.py')
-rw-r--r--rumba/utils.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/rumba/utils.py b/rumba/utils.py
index 88ea0d3..822deb3 100644
--- a/rumba/utils.py
+++ b/rumba/utils.py
@@ -29,6 +29,7 @@ import os
import rumba.log as log
import rumba.model as model
+import rumba.testbeds.local as local
# Fix input reordering
try:
@@ -204,7 +205,11 @@ class ExperimentManager(object):
# Swap out
if do_swap_out:
- self.experiment.swap_out()
+ # Kill prototype in case of local testbed
+ if isinstance(self.experiment.testbed, local.Testbed):
+ self.experiment.terminate_prototype()
+ else:
+ self.experiment.swap_out()
if exc_val is not None:
logger.error('Something went wrong. '
'Got %s: %s',