Skip to content

Commit b69e9c5

Browse files
Editing deployments with SDK (#97)
* add update endpoints to client sdk + upgrade api client version * update version number * upgrade mypy for tests
1 parent 6d6dd7d commit b69e9c5

File tree

4 files changed

+12
-3
lines changed

4 files changed

+12
-3
lines changed

centml/sdk/api.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,15 @@ def create_compute(self, request: CreateComputeDeploymentRequest):
4343
def create_cserve(self, request: CreateCServeV2DeploymentRequest):
4444
return self._api.create_cserve_v2_deployment_deployments_cserve_v2_post(request)
4545

46+
def update_inference(self, deployment_id: int, request: CreateInferenceDeploymentRequest):
47+
return self._api.update_inference_deployment_deployments_inference_put(deployment_id, request)
48+
49+
def update_compute(self, deployment_id: int, request: CreateComputeDeploymentRequest):
50+
return self._api.update_compute_deployment_deployments_compute_put(deployment_id, request)
51+
52+
def update_cserve(self, deployment_id: int, request: CreateCServeV2DeploymentRequest):
53+
return self._api.update_cserve_v2_deployment_deployments_cserve_v2_put(deployment_id, request)
54+
4655
def _update_status(self, id, new_status):
4756
status_req = platform_api_python_client.DeploymentStatusRequest(status=new_status)
4857
self._api.update_deployment_status_deployments_status_deployment_id_put(id, status_req)

requirements-dev.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,6 @@ transformers>=4.34.1
99
httpx>=0.25.0
1010
Pillow>=10.1.0
1111
parameterized>=0.9.0
12-
mypy==1.5.1
12+
mypy==1.15.0
1313
types-requests==2.31.0.2
1414
types-tabulate>=0.9.0

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,4 +9,4 @@ cryptography==44.0.1
99
prometheus-client>=0.20.0
1010
scipy>=1.6.0
1111
scikit-learn>=1.5.1
12-
platform-api-python-client==3.2.4
12+
platform-api-python-client==3.2.7

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
setup(
1313
name='centml',
14-
version='0.3.4',
14+
version='0.3.5',
1515
packages=find_packages(),
1616
python_requires=">=3.10",
1717
long_description=open('README.md').read(),

0 commit comments

Comments
 (0)