diff options
-rw-r--r-- | rumba/prototypes/irati_templates.py | 2 | ||||
-rw-r--r-- | rumba/storyboard.py | 4 |
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): """ |