update to zip result

swiss-ai-center · Jan 19, 2024 · f785b63 · f785b63
1 parent de4d90d
commit f785b63
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/main.py b/src/main.py
@@ -64,7 +64,9 @@ def __init__(self):
                 ),
             ],
             data_out_fields=[
-                FieldDescription(name="result", type=[FieldDescriptionType.TEXT_PLAIN]),
+                FieldDescription(
+                    name="result", type=[FieldDescriptionType.APPLICATION_ZIP]
+                ),
             ],
             tags=[
                 ExecutionUnitTag(
@@ -120,7 +122,7 @@ def process(self, data):
                     )
         return {
             "result": TaskData(
-                data=zip_buffer.getvalue(), type=FieldDescriptionType.TEXT_PLAIN
+                data=zip_buffer.getvalue(), type=FieldDescriptionType.APPLICATION_ZIP
             )
         }