Skip to content

Commit 9332d2b

Browse files
authored
Replace DeploymentState with DeploymentAssignmentState (#2942)
1 parent 3c75c87 commit 9332d2b

File tree

3 files changed

+32
-55
lines changed

3 files changed

+32
-55
lines changed

output/schema/schema.json

Lines changed: 17 additions & 35 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

output/typescript/types.ts

Lines changed: 2 additions & 4 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

specification/ml/_types/TrainedModel.ts

Lines changed: 13 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ export class TrainedModelDeploymentStats {
9393
/** The epoch timestamp when the deployment started. */
9494
start_time: EpochTime<UnitMillis>
9595
/** The overall state of the deployment. */
96-
state: DeploymentState
96+
state: DeploymentAssignmentState
9797
/** The number of threads used be each allocation during inference. */
9898
threads_per_allocation: integer
9999
/** The sum of `timeout_count` for all nodes in the deployment. */
@@ -270,40 +270,37 @@ export enum TrainedModelType {
270270
pytorch
271271
}
272272

273-
export enum DeploymentState {
273+
export enum DeploymentAllocationState {
274274
/**
275-
* The deployment is usable; at least one node has the model allocated.
275+
* The trained model is started on at least one node.
276276
*/
277277
started,
278278
/**
279-
* The deployment has recently started but is not yet usable; the model is not allocated on any nodes.
279+
* Trained model deployment is starting but it is not yet deployed on any nodes.
280280
*/
281281
starting,
282282
/**
283-
* The deployment is preparing to stop and deallocate the model from the relevant nodes.
283+
* Trained model deployment has started on all valid nodes.
284284
*/
285-
stopping
285+
fully_allocated
286286
}
287287

288-
export enum DeploymentAllocationState {
288+
export enum DeploymentAssignmentState {
289289
/**
290-
* The trained model is started on at least one node.
290+
* The deployment is usable; at least one node has the model allocated.
291291
*/
292292
started,
293293
/**
294-
* Trained model deployment is starting but it is not yet deployed on any nodes.
294+
* The deployment has recently started but is not yet usable; the model is not allocated on any nodes.
295295
*/
296296
starting,
297297
/**
298-
* Trained model deployment has started on all valid nodes.
298+
* The deployment is preparing to stop and deallocate the model from the relevant nodes.
299299
*/
300-
fully_allocated
301-
}
302-
303-
export enum DeploymentAssignmentState {
304-
starting,
305-
started,
306300
stopping,
301+
/**
302+
* The deployment is on a failed state and must be re-deployed.
303+
*/
307304
failed
308305
}
309306

0 commit comments

Comments
 (0)