python: remove internal `import bt2` imports

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

diff --git a/src/bindings/python/bt2/bt2/trace.py b/src/bindings/python/bt2/bt2/trace.py
index 5cbe82c64d6eda6f1e996a2b2071ede0bdab4cd8..32e730e36bb464e750c6bf12d3b4e1e3a78ddde4 100644 (file)
--- a/src/bindings/python/bt2/bt2/trace.py
+++ b/src/bindings/python/bt2/bt2/trace.py
@@ -9,7 +9,8 @@ import collections.abc
 from bt2 import value as bt2_value
 from bt2 import stream as bt2_stream
 from bt2 import stream_class as bt2_stream_class
-import bt2
+from bt2 import error as bt2_error
+
 import functools
 import uuid as uuidp
 
@@ -246,7 +247,7 @@ class _Trace(_TraceConst):
             )
 
         if stream_ptr is None:
-            raise bt2._MemoryError("cannot create stream object")
+            raise bt2_error._MemoryError("cannot create stream object")
 
         stream = bt2_stream._Stream._create_from_ptr(stream_ptr)