From 031cdc38878686d7ae5532f3968e2b4d428ab7cd Mon Sep 17 00:00:00 2001 From: Sander Vrijders Date: Wed, 9 May 2018 15:50:51 +0200 Subject: elements: Allow exporting to DMS YAML This allows for exporting the experiment to a YAML format which is parsable by the DMS. --- rumba/elements/experimentation.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'rumba/elements/experimentation.py') diff --git a/rumba/elements/experimentation.py b/rumba/elements/experimentation.py index 9890975..2c6f478 100644 --- a/rumba/elements/experimentation.py +++ b/rumba/elements/experimentation.py @@ -521,6 +521,15 @@ class Experiment(object): end = time.time() logger.info("Swap-out took %.2f seconds", end - start) + def to_dms_yaml(self, filename, clean=True): + mode = 'w' + if not clean: + mode += '+' + with open(filename, mode) as f: + for node in self.nodes: + f.write("---\n") + node.to_dms_yaml(f) + f.write("...\n") class Executor: __metaclass__ = abc.ABCMeta -- cgit v1.2.3