Skip to content

Commit 652c548

Browse files
authored
fix(ray): fix semantic segmentation task parser (#237)
Because - semantic segmentation task output parse does not follow task schema This commit - fix output key
1 parent 019a775 commit 652c548

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

instill/helpers/ray_io.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -321,7 +321,7 @@ def construct_task_semantic_segmentation_output(
321321
for r, cat in zip(rle, category):
322322
objects.append({"rle": str(r), "category": str(cat)})
323323

324-
data["objects"] = objects
324+
data["stuffs"] = objects
325325

326326
if isinstance(request, Request):
327327
task_outputs.append({"data": data})

0 commit comments

Comments
 (0)