Skip to content
Merged
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
10 changes: 0 additions & 10 deletions centml/sdk/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,6 @@
DeploymentStatus,
CreateInferenceV3DeploymentRequest,
CreateComputeDeploymentRequest,
CreateCServeV3DeploymentRequest,
CreateDynamoDeploymentRequest,
CreateJobDeploymentRequest,
CreateHardwareInstanceRequest,
Expand Down Expand Up @@ -169,9 +168,6 @@ def create_compute(self, request: CreateComputeDeploymentRequest):
def create_job(self, request: CreateJobDeploymentRequest):
return self._api.create_job_deployment_deployments_job_post(request)

def create_cserve(self, request: CreateCServeV3DeploymentRequest):
return self._api.create_cserve_v3_deployment_deployments_cserve_v3_post(request)

def create_dynamo(self, request: CreateDynamoDeploymentRequest):
return self._api.create_dynamo_deployment_deployments_dynamo_post(request)

Expand All @@ -181,9 +177,6 @@ def update_inference(self, deployment_id: int, request: CreateInferenceV3Deploym
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: CreateCServeV3DeploymentRequest):
return self._api.update_cserve_v3_deployment_deployments_cserve_v3_put(deployment_id, request)

def update_dynamo(self, deployment_id: int, request: CreateDynamoDeploymentRequest):
return self._api.update_dynamo_deployment_deployments_dynamo_put(deployment_id, request)

Expand Down Expand Up @@ -220,9 +213,6 @@ def delete_hardware_instance(self, hardware_instance_id: int):
def get_prebuilt_images(self, depl_type: DeploymentType):
return self._api.get_prebuilt_images_prebuilt_images_get(type=depl_type)

def get_cserve_recipe(self, model=None, hf_token=None):
return self._api.get_cserve_recipe_deployments_cserve_recipes_get(model=model, hf_token=hf_token).results

def get_cluster_id(self, hardware_instance_id):
filtered_hw = list(filter(lambda h: h.id == hardware_instance_id, self.get_hardware_instances()))

Expand Down
73 changes: 0 additions & 73 deletions examples/sdk/create_cserve.py

This file was deleted.

2 changes: 1 addition & 1 deletion tests/test_cli_cluster.py
Original file line number Diff line number Diff line change
Expand Up @@ -158,7 +158,7 @@ def test_get_job_routes_to_job_api_and_displays_job_config():

deployment = _deployment(status=DeploymentStatus.ACTIVE)
hardware = HardwareInstanceResponse(
id=2, name="h100", gpu_type="H100", num_gpu=8, cpu=96, memory=1024, cluster_id=1
id=2, name="h100", gpu_type="H100", num_gpu=8, cpu=96, memory=1024, cost_per_hr=0, cluster_id=1
)
runner = CliRunner()

Expand Down
Loading