Python: examples: import explicit BT modules

[babeltrace.git] / bindings / python / examples / babeltrace_and_lttng.py

diff --git a/bindings/python/examples/babeltrace_and_lttng.py b/bindings/python/examples/babeltrace_and_lttng.py
index 0b92c0f6d79cfde7c665ec8a9dcc7205576fcd28..651fa2da982a81ec06fc0c0aaeb92f074a884935 100644 (file)
--- a/bindings/python/examples/babeltrace_and_lttng.py
+++ b/bindings/python/examples/babeltrace_and_lttng.py
@@ -45,7 +45,7 @@ out_file = "babeltrace-lttng-trace-text-output.txt"
 
 import time
 try:
-    import babeltrace
+    import babeltrace.reader
     import lttng
 except ImportError:
     raise ImportError( "both babeltrace and lttng-tools "
@@ -109,7 +109,7 @@ if ret < 0:
 # BABELTRACE
 
 # Create TraceCollecion and add trace:
-traces = babeltrace.TraceCollection()
+traces = babeltrace.reader.TraceCollection()
 ret = traces.add_trace(trace_path + "/kernel", "ctf")
 if ret is None:
     raise BabeltraceError("Error adding trace")