fixed small bug

IBM · Sep 21, 2024 · d8f531a · d8f531a
1 parent 2b44b4c
commit d8f531a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/data-processing-lib/python/src/data_processing/transform/pipeline_transform.py b/data-processing-lib/python/src/data_processing/transform/pipeline_transform.py
@@ -94,7 +94,10 @@ def _convert_output(data: list[tuple[bytes, str]]) -> list[tuple[bytes, str]]:
         i = 0
         for dt in data:
             fname = TransformUtils.get_file_extension(dt[1])
-            res[i] = (dt[0], fname[1])
+            ext = fname[1]
+            if len(ext) <= 1:
+                ext = fname[0]
+            res[i] = (dt[0], ext)
             i += 1
         return res