Coverage for backend/common/models/idaes/payloads/ml_request_schema.py: 100%
18 statements
« prev ^ index » next coverage.py v7.10.7, created at 2025-11-06 23:27 +0000
2from typing import Optional
3from pydantic import BaseModel
4from common.models import CloudEventsPayload
5class MLTrainRequestPayload(BaseModel):
6 """request schema"""
7 datapoints: list
8 columns: list[str]
9 input_labels: list[str]
10 output_labels: list[str]
11 task_id: int
13class MLTrainingCompletionPayload(BaseModel):
14 """response schema"""
15 json_response: dict
16 error: Optional[str] # error message if applicable
17 log: Optional[str] # log
18 traceback: Optional[str] # traceback if applicable
19 task_id: int
20 status: str
22MLTrainingEvent = CloudEventsPayload[MLTrainRequestPayload]
23MLTrainingCompletionEvent = CloudEventsPayload[MLTrainingCompletionPayload]