python: remove internal `import bt2` imports

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

diff --git a/src/bindings/python/bt2/bt2/trace_class.py b/src/bindings/python/bt2/bt2/trace_class.py
index c25154e8f9d9f4fb1584c26f5a83638a3d317d88..f6efe71c2135ac224cbfa8851222bc1dfc97c0cf 100644 (file)
--- a/src/bindings/python/bt2/bt2/trace_class.py
+++ b/src/bindings/python/bt2/bt2/trace_class.py
@@ -11,10 +11,10 @@ from bt2 import stream_class as bt2_stream_class
 from bt2 import field_class as bt2_field_class
 from bt2 import integer_range_set as bt2_integer_range_set
 from bt2 import trace as bt2_trace
+from bt2 import error as bt2_error
 from bt2 import value as bt2_value
 import collections.abc
 import functools
-import bt2
 
 
 def _trace_class_destruction_listener_from_native(
@@ -149,7 +149,7 @@ class _TraceClass(_TraceClassConst):
         trace_ptr = native_bt.trace_create(self._ptr)
 
         if trace_ptr is None:
-            raise bt2._MemoryError("cannot create trace class object")
+            raise bt2_error._MemoryError("cannot create trace class object")
 
         trace = bt2_trace._Trace._create_from_ptr(trace_ptr)
 
@@ -281,7 +281,9 @@ class _TraceClass(_TraceClassConst):
 
     def _check_field_class_create_status(self, ptr, type_name):
         if ptr is None:
-            raise bt2._MemoryError("cannot create {} field class".format(type_name))
+            raise bt2_error._MemoryError(
+                "cannot create {} field class".format(type_name)
+            )
 
     @staticmethod
     def _set_field_class_user_attrs(fc, user_attributes):