diff --git a/inference/core/workflows/core_steps/models/foundation/seg_preview/v1.py b/inference/core/workflows/core_steps/models/foundation/seg_preview/v1.py index ff94688c70..00ae1db8be 100644 --- a/inference/core/workflows/core_steps/models/foundation/seg_preview/v1.py +++ b/inference/core/workflows/core_steps/models/foundation/seg_preview/v1.py @@ -42,6 +42,13 @@ WorkflowBlockManifest, ) +_OUTPUTS = [ + OutputDefinition( + name="predictions", + kind=[INSTANCE_SEGMENTATION_PREDICTION_KIND], + ), +] + DETECTIONS_CLASS_NAME_FIELD = "class_name" DETECTION_ID_FIELD = "detection_id" @@ -93,12 +100,7 @@ def get_parameters_accepting_batches(cls) -> List[str]: @classmethod def describe_outputs(cls) -> List[OutputDefinition]: - return [ - OutputDefinition( - name="predictions", - kind=[INSTANCE_SEGMENTATION_PREDICTION_KIND], - ), - ] + return _OUTPUTS @classmethod def get_execution_engine_compatibility(cls) -> Optional[str]: