aboutsummaryrefslogtreecommitdiff
path: root/examples/jfed-rlite.py
diff options
context:
space:
mode:
Diffstat (limited to 'examples/jfed-rlite.py')
-rwxr-xr-xexamples/jfed-rlite.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/examples/jfed-rlite.py b/examples/jfed-rlite.py
index d80b56e..5cc087f 100755
--- a/examples/jfed-rlite.py
+++ b/examples/jfed-rlite.py
@@ -1,6 +1,7 @@
#!/usr/bin/env python
from rumba.model import *
+from rumba.utils import ExperimentManager
import rumba.testbeds.jfed as jfed
import rumba.prototypes.rlite as rlite
@@ -46,9 +47,7 @@ tb = jfed.Testbed(exp_name = args.expname,
exp = rlite.Experiment(tb, nodes = [a, b])
-try:
+with ExperimentManager(exp):
exp.swap_in()
exp.install_prototype()
exp.bootstrap_prototype()
-finally:
- exp.swap_out()