ground_truth: better error handling when logging discarded flows

stratosphereips · Nov 5, 2024 · 24b2e71 · 24b2e71
1 parent e9641f4
commit 24b2e71
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/parsers/ground_truth.py b/parsers/ground_truth.py
@@ -256,7 +256,8 @@ def extract_fields(self, line: str) -> Tuple[Union[bool,dict], str]:
 
         try:
             if flow[0] == "unknown":
-                return False, f"Unsupported flow label '{flow[0]}'"
+                label = "background" if "Background" in line else ""
+                return False, f"Unsupported flow label {label}"
 
             return {
                'label':  flow[0],
@@ -396,7 +397,7 @@ def parse_file(self, filename: str):
 
             flow, err = self.extract_fields(line)
             if not flow:
-                self.log(f"{err}. Skipping flow at line",
+                self.log(f"{err}. Skipping flow at line: ",
                          line_number,
                          error=True)
                 continue