diff options
Diffstat (limited to 'examples/snake.py')
| -rwxr-xr-x | examples/snake.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/examples/snake.py b/examples/snake.py index c6b67b3..6f9c48d 100755 --- a/examples/snake.py +++ b/examples/snake.py @@ -3,6 +3,7 @@ # An example script using the rumba package from rumba.model import * +from rumba.utils import ExperimentManager # import testbed plugins import rumba.testbeds.emulab as emulab @@ -18,6 +19,7 @@ import rumba.prototypes.irati as irati import argparse import sys + description = "Script to create a snake" argparser = argparse.ArgumentParser(description = description) @@ -62,8 +64,6 @@ exp = rl.Experiment(tb, nodes = nodes) print(exp) -try: +with ExperimentManager(exp): exp.swap_in() exp.bootstrap_prototype() -finally: - exp.swap_out() |