aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--rumba/prototypes/irati_templates.py2
-rw-r--r--rumba/storyboard.py4
2 files changed, 3 insertions, 3 deletions
diff --git a/rumba/prototypes/irati_templates.py b/rumba/prototypes/irati_templates.py
index 6e5932d..a6dd3c4 100644
--- a/rumba/prototypes/irati_templates.py
+++ b/rumba/prototypes/irati_templates.py
@@ -369,7 +369,7 @@ policy_translator = {
'routing': lambda d, v, p: ps_set(
d["routingConfiguration"], "policySet", v, p),
'resource-allocator.pduftg': lambda d, v, p: ps_set(
- d["resourceAllocatorConfiguration"], "policySet", v, p),
+ d["resourceAllocatorConfiguration"]["pduftgConfiguration"], "policySet", v, p),
'efcp.*.dtcp': None,
'efcp.*.dtp': None,
}
diff --git a/rumba/storyboard.py b/rumba/storyboard.py
index f03a2e8..200cca7 100644
--- a/rumba/storyboard.py
+++ b/rumba/storyboard.py
@@ -45,7 +45,7 @@ import rumba.log as log
try:
from io import StringIO
except ImportError:
- import StringIO # Python 2 here
+ from StringIO import StringIO # Python 2 here
logger = log.get_logger(__name__)
@@ -1221,7 +1221,7 @@ class StoryBoard(_SBEntity):
"""
s = StringIO()
self.write_script(s)
- return s
+ return s.getvalue()
def parse_script(self, buffer, clean=True):
"""