@@ -48,32 +48,7 @@ async def submit_training_job(
48
48
tags : List [str ],
49
49
filter : Optional [Filter ] = None ,
50
50
) -> str :
51
- """Submits a training job to be processed by the training job manager.
52
-
53
- Args:
54
- org_id (str): the ID of the organization whose data will be used for training.
55
- model_name (str): the training model name.
56
- model_version (str): the training model version.
57
- model_type (viam.proto.app.mltraining.ModelType): the training model type.
58
- tags (List[str]): list of tags to apply to the training model output.
59
- filter (Optional[viam.proto.app.data.Filter]): `Filter` specifying which data to include when training.
60
- If unspecified, all data will be included.
61
-
62
- Returns:
63
- str: the id assigned to the training job.
64
- """
65
-
66
- request = SubmitTrainingJobRequest (
67
- organization_id = org_id ,
68
- model_name = model_name ,
69
- model_version = model_version ,
70
- model_type = model_type ,
71
- tags = tags ,
72
- filter = filter ,
73
- )
74
- response : SubmitTrainingJobResponse = await self ._ml_training_client .SubmitTrainingJob (request , metadata = self ._metadata )
75
-
76
- return response .id
51
+ raise NotImplementedError ()
77
52
78
53
async def get_training_job (self , id : str ) -> TrainingJobMetadata :
79
54
"""Gets training job data.
@@ -117,6 +92,9 @@ async def cancel_training_job(self, id: str) -> None:
117
92
118
93
Args:
119
94
id (str): the id of the job to be canceled.
95
+
96
+ Raises:
97
+ GRPCError: if no training job exists with the given id.
120
98
"""
121
99
122
100
request = CancelTrainingJobRequest (id = id )
0 commit comments