Skip to content

Commit

Permalink
Add ml.update_trained_model_deployment API (#2564)
Browse files Browse the repository at this point in the history
(cherry picked from commit 4195183)
  • Loading branch information
pquentin authored and github-actions[bot] committed May 24, 2024
1 parent 4279c2b commit 01874d5
Show file tree
Hide file tree
Showing 5 changed files with 182 additions and 10 deletions.
99 changes: 95 additions & 4 deletions output/schema/schema.json

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

6 changes: 0 additions & 6 deletions output/schema/validation-errors.json

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

11 changes: 11 additions & 0 deletions output/typescript/types.ts

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

Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
/*
* Licensed to Elasticsearch B.V. under one or more contributor
* license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright
* ownership. Elasticsearch B.V. licenses this file to you under
* the Apache License, Version 2.0 (the "License"); you may
* not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/

import { RequestBase } from '@_types/Base'
import { Id } from '@_types/common'
import { integer } from '@_types/Numeric'

/**
* Starts a trained model deployment, which allocates the model to every machine learning node.
* @rest_spec_name ml.update_trained_model_deployment
* @availability stack since=8.6.0 stability=stable
* @availability serverless stability=beta visibility=public
* @cluster_privileges manage_ml
*/
export interface Request extends RequestBase {
path_parts: {
/**
* The unique identifier of the trained model. Currently, only PyTorch models are supported.
*/
model_id: Id
}
body: {
/**
* The number of model allocations on each node where the model is deployed.
* All allocations on a node share the same copy of the model in memory but use
* a separate set of threads to evaluate the model.
* Increasing this value generally increases the throughput.
* If this setting is greater than the number of hardware threads
* it will automatically be changed to a value less than the number of hardware threads.
* @server_default 1
*/
number_of_allocations?: integer
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
/*
* Licensed to Elasticsearch B.V. under one or more contributor
* license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright
* ownership. Elasticsearch B.V. licenses this file to you under
* the Apache License, Version 2.0 (the "License"); you may
* not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/

import { TrainedModelAssignment } from '../_types/TrainedModel'

export class Response {
body: {
assignment: TrainedModelAssignment
}
}

0 comments on commit 01874d5

Please sign in to comment.