bt2: import public names into `__init__.py`

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

diff --git a/src/bindings/python/bt2/bt2/stream_class.py b/src/bindings/python/bt2/bt2/stream_class.py
index 08186333f1257f7591edcfe367d09f062c47b3a4..cda21864b5c409b515a611275debcc02f06820a3 100644 (file)
--- a/src/bindings/python/bt2/bt2/stream_class.py
+++ b/src/bindings/python/bt2/bt2/stream_class.py
@@ -23,6 +23,7 @@
 from bt2 import native_bt, object, utils
 import bt2.field_class
 import bt2.event_class
+import bt2.trace_class
 import collections.abc
 import bt2.stream
 import bt2
@@ -107,7 +108,7 @@ class _StreamClass(object._SharedObject, collections.abc.Mapping):
         tc_ptr = native_bt.stream_class_borrow_trace_class_const(self._ptr)
 
         if tc_ptr is not None:
-            return bt2._TraceClass._create_from_ptr_and_get_ref(tc_ptr)
+            return bt2.trace_class._TraceClass._create_from_ptr_and_get_ref(tc_ptr)
 
     @property
     def name(self):