bt2: import public names into `__init__.py`

[babeltrace.git] / src / bindings / python / bt2 / bt2 / component.py

diff --git a/src/bindings/python/bt2/bt2/component.py b/src/bindings/python/bt2/bt2/component.py
index 3557445681c0cb5dee1adb43cced1a0205888e4c..7a79666abae5455db3ce36ed7e2e76b83fd6e3da 100644 (file)
--- a/src/bindings/python/bt2/bt2/component.py
+++ b/src/bindings/python/bt2/bt2/component.py
@@ -24,6 +24,7 @@ from bt2 import native_bt, object, utils
 import bt2.message_iterator
 import collections.abc
 import bt2.value
+import bt2.trace_class
 import traceback
 import bt2.port
 import sys
@@ -679,7 +680,7 @@ class _UserComponent(metaclass=_UserComponentType):
         if tc_ptr is None:
             raise bt2._MemoryError('could not create trace class')
 
-        tc = bt2._TraceClass._create_from_ptr(tc_ptr)
+        tc = bt2.trace_class._TraceClass._create_from_ptr(tc_ptr)
         tc._assigns_automatic_stream_class_id = assigns_automatic_stream_class_id
 
         return tc