diff --git a/centml/sdk/api.py b/centml/sdk/api.py index 1120fdb..5d0042f 100644 --- a/centml/sdk/api.py +++ b/centml/sdk/api.py @@ -43,6 +43,15 @@ def create_compute(self, request: CreateComputeDeploymentRequest): def create_cserve(self, request: CreateCServeV2DeploymentRequest): return self._api.create_cserve_v2_deployment_deployments_cserve_v2_post(request) + def update_inference(self, deployment_id: int, request: CreateInferenceDeploymentRequest): + return self._api.update_inference_deployment_deployments_inference_put(deployment_id, request) + + def update_compute(self, deployment_id: int, request: CreateComputeDeploymentRequest): + return self._api.update_compute_deployment_deployments_compute_put(deployment_id, request) + + def update_cserve(self, deployment_id: int, request: CreateCServeV2DeploymentRequest): + return self._api.update_cserve_v2_deployment_deployments_cserve_v2_put(deployment_id, request) + def _update_status(self, id, new_status): status_req = platform_api_python_client.DeploymentStatusRequest(status=new_status) self._api.update_deployment_status_deployments_status_deployment_id_put(id, status_req) diff --git a/requirements-dev.txt b/requirements-dev.txt index 6179c20..6f00c8b 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -9,6 +9,6 @@ transformers>=4.34.1 httpx>=0.25.0 Pillow>=10.1.0 parameterized>=0.9.0 -mypy==1.5.1 +mypy==1.15.0 types-requests==2.31.0.2 types-tabulate>=0.9.0 diff --git a/requirements.txt b/requirements.txt index ff87c33..9c4ff41 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,4 +9,4 @@ cryptography==44.0.1 prometheus-client>=0.20.0 scipy>=1.6.0 scikit-learn>=1.5.1 -platform-api-python-client==3.2.4 +platform-api-python-client==3.2.7 diff --git a/setup.py b/setup.py index fb91703..eb36db4 100644 --- a/setup.py +++ b/setup.py @@ -11,7 +11,7 @@ setup( name='centml', - version='0.3.4', + version='0.3.5', packages=find_packages(), python_requires=">=3.10", long_description=open('README.md').read(),