Skip to content

Fix ml for validation #2952

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 2 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
52 changes: 17 additions & 35 deletions output/schema/schema.json

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 2 additions & 4 deletions output/typescript/types.ts

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

29 changes: 13 additions & 16 deletions specification/ml/_types/TrainedModel.ts
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,7 @@ export class TrainedModelDeploymentStats {
/** The epoch timestamp when the deployment started. */
start_time: EpochTime<UnitMillis>
/** The overall state of the deployment. */
state: DeploymentState
state: DeploymentAssignmentState
/** The number of threads used be each allocation during inference. */
threads_per_allocation: integer
/** The sum of `timeout_count` for all nodes in the deployment. */
Expand Down Expand Up @@ -270,40 +270,37 @@ export enum TrainedModelType {
pytorch
}

export enum DeploymentState {
export enum DeploymentAllocationState {
/**
* The deployment is usable; at least one node has the model allocated.
* The trained model is started on at least one node.
*/
started,
/**
* The deployment has recently started but is not yet usable; the model is not allocated on any nodes.
* Trained model deployment is starting but it is not yet deployed on any nodes.
*/
starting,
/**
* The deployment is preparing to stop and deallocate the model from the relevant nodes.
* Trained model deployment has started on all valid nodes.
*/
stopping
fully_allocated
}

export enum DeploymentAllocationState {
export enum DeploymentAssignmentState {
/**
* The trained model is started on at least one node.
* The deployment is usable; at least one node has the model allocated.
*/
started,
/**
* Trained model deployment is starting but it is not yet deployed on any nodes.
* The deployment has recently started but is not yet usable; the model is not allocated on any nodes.
*/
starting,
/**
* Trained model deployment has started on all valid nodes.
* The deployment is preparing to stop and deallocate the model from the relevant nodes.
*/
fully_allocated
}

export enum DeploymentAssignmentState {
starting,
started,
stopping,
/**
* The deployment is on a failed state and must be re-deployed.
*/
failed
}

Expand Down