Fix unconsumed stream error

liiatools/csww_pipeline/stream_record.py

@@ -1,33 +1,35 @@
+from enum import Enum
 from typing import Iterator
 
 import pandas as pd
 import tablib
 from more_itertools import peekable
-from enum import Enum
-
 from sfdata_stream_parser import events
 from sfdata_stream_parser.collectors import xml_collector
-from sfdata_stream_parser.filters.generic import (generator_with_value)
+from sfdata_stream_parser.filters.generic import generator_with_value
 
 
 class CSWWEvent(events.ParseEvent):
     @staticmethod
     def name():
         return "Worker"
+
     pass
 
 
 class LALevelEvent(events.ParseEvent):
     @staticmethod
     def name():
         return "LA_Level"
+
     pass
 
 
 class HeaderEvent(events.ParseEvent):
     @staticmethod
     def name():
         return "Header"
+
     pass
 
 
@@ -59,7 +61,6 @@ def text_collector(stream):
     return _reduce_dict(data_dict)
 
 
-@xml_collector
 def message_collector(stream):
     """
     Collect messages from XML elements and yield events