No longer return model_id in record

model_server/base/pipelines/params.py

@@ -33,6 +33,5 @@ class PipelineParams(BaseModel):
 class PipelineRecord(BaseModel):
     output_accessor_id: str
     interm_accessor_ids: Union[List[str], None]
-    model_id: str
     success: bool
     timer: dict

model_server/base/pipelines/segment.py

@@ -16,14 +16,19 @@ router = APIRouter(
     tags=['pipelines'],
 )
 
+
 class SegmentParams(PipelineParams):
     accessor_id: str = Field(description='ID(s) of previously loaded accessor(s) to use as pipeline input')
     model_id: str = Field(description='ID(s) of previously loaded segmentation model(s)')
     channel: int = Field(None, description='Channel to use for segmentation; use all channels if empty.')
 
 
+class SegmentRecord(PipelineRecord):
+    pass
+
+
 @router.put('/segment')
-def segment(p: SegmentParams) -> PipelineRecord:
+def segment(p: SegmentParams) -> SegmentRecord:
     """
     Run a semantic segmentation model to compute a binary mask from an input image
     """