Skip to content

Commit ed359e8

Browse files
committed
unimplement submit_training_job
1 parent f7dda82 commit ed359e8

File tree

1 file changed

+4
-26
lines changed

1 file changed

+4
-26
lines changed

src/viam/app/ml_training_client.py

Lines changed: 4 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -48,32 +48,7 @@ async def submit_training_job(
4848
tags: List[str],
4949
filter: Optional[Filter] = None,
5050
) -> 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()
7752

7853
async def get_training_job(self, id: str) -> TrainingJobMetadata:
7954
"""Gets training job data.
@@ -117,6 +92,9 @@ async def cancel_training_job(self, id: str) -> None:
11792
11893
Args:
11994
id (str): the id of the job to be canceled.
95+
96+
Raises:
97+
GRPCError: if no training job exists with the given id.
12098
"""
12199

122100
request = CancelTrainingJobRequest(id=id)

0 commit comments

Comments
 (0)