doc: Make python bindings doc build

[babeltrace.git] / bindings / python / babeltrace / babeltrace / reader_trace_collection.py

diff --git a/bindings/python/babeltrace/babeltrace/reader_trace_collection.py b/bindings/python/babeltrace/babeltrace/reader_trace_collection.py
index d888f8d811ffb7c4e80b0bc2b1fbae82d0fd8619..e6544a8229dd3eebd3ea906856ba531e79808fa8 100644 (file)
--- a/bindings/python/babeltrace/babeltrace/reader_trace_collection.py
+++ b/bindings/python/babeltrace/babeltrace/reader_trace_collection.py
@@ -166,14 +166,14 @@ class TraceCollection:
         return self._gen_events(timestamp_begin / 1e9, timestamp_end / 1e9)
 
     def _gen_events(self, begin_s=None, end_s=None):
-        specs = [bt2.SourceComponentSpec('ctf', 'fs', th.path) for th in self._trace_handles]
+        specs = [bt2.ComponentSpec('ctf', 'fs', th.path) for th in self._trace_handles]
 
         try:
-            iter_cls = bt2.TraceCollectionNotificationIterator
+            iter_cls = bt2.TraceCollectionMessageIterator
             tc_iter = iter_cls(specs,
                                stream_intersection_mode=self._intersect_mode,
                                begin=begin_s, end=end_s,
-                               notification_types=[bt2.EventNotification])
+                               message_types=[bt2.EventMessage])
             return map(reader_event._create_event, tc_iter)
         except:
             raise ValueError