X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=src%2Fbindings%2Fpython%2Fbt2%2Fbt2%2Fport.py;h=ecee18ce47b7eaa65dca8a218136572f9cd26c2a;hb=4acc866e829881299a13d0aa8e28b93807549975;hp=4ba59f066df4c230e4b3ef92389f1ae7b50ad102;hpb=d24d56638469189904fb6ddbb3c725817b3e9417;p=babeltrace.git

diff --git a/src/bindings/python/bt2/bt2/port.py b/src/bindings/python/bt2/bt2/port.py
index 4ba59f06..ecee18ce 100644
--- a/src/bindings/python/bt2/bt2/port.py
+++ b/src/bindings/python/bt2/bt2/port.py
@@ -110,18 +110,24 @@ class _UserComponentPort(_Port):
 
 
 class _UserComponentInputPort(_UserComponentPort, _InputPort):
-    _as_self_port_ptr = staticmethod(native_bt.self_component_port_input_as_self_component_port)
+    _as_self_port_ptr = staticmethod(
+        native_bt.self_component_port_input_as_self_component_port
+    )
 
     def create_message_iterator(self):
-        msg_iter_ptr = native_bt.self_component_port_input_message_iterator_create(self._ptr)
+        msg_iter_ptr = native_bt.self_component_port_input_message_iterator_create(
+            self._ptr
+        )
         if msg_iter_ptr is None:
-            raise bt2.CreationError('cannot create message iterator object')
+            raise bt2.MemoryError('cannot create message iterator object')
 
         return bt2.message_iterator._UserComponentInputPortMessageIterator(msg_iter_ptr)
 
 
 class _UserComponentOutputPort(_UserComponentPort, _OutputPort):
-    _as_self_port_ptr = staticmethod(native_bt.self_component_port_output_as_self_component_port)
+    _as_self_port_ptr = staticmethod(
+        native_bt.self_component_port_output_as_self_component_port
+    )
 
 
 _PORT_TYPE_TO_PYCLS = {