From 240f052a67e4567a814a3e994cfcdc0645c82908 Mon Sep 17 00:00:00 2001
From: Kaj Siebert <kaj@k-si.com>
Date: Wed, 29 Nov 2023 18:57:01 +0100
Subject: [PATCH] Fix unconsumed stream error

---
 liiatools/csww_pipeline/stream_record.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/liiatools/csww_pipeline/stream_record.py b/liiatools/csww_pipeline/stream_record.py
index 412009cd..bdf94cad 100644
--- a/liiatools/csww_pipeline/stream_record.py
+++ b/liiatools/csww_pipeline/stream_record.py
@@ -1,19 +1,19 @@
+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
 
 
@@ -21,6 +21,7 @@ class LALevelEvent(events.ParseEvent):
     @staticmethod
     def name():
         return "LA_Level"
+
     pass
 
 
@@ -28,6 +29,7 @@ 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