Skip to content

Commit 9e25528

Browse files
committed
format lint
1 parent 1d43068 commit 9e25528

File tree

2 files changed

+12
-12
lines changed

2 files changed

+12
-12
lines changed

centml/cli/cluster.py

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -277,11 +277,7 @@ def create():
277277
k, v = kv.strip().split("=")
278278
env_vars[k] = v
279279

280-
command_str = click.prompt(
281-
"Enter command (space-separated) or leave blank",
282-
default="",
283-
show_default=False
284-
)
280+
command_str = click.prompt("Enter command (space-separated) or leave blank", default="", show_default=False)
285281

286282
command = command_str.strip() if command_str.strip() else None
287283

@@ -429,13 +425,14 @@ def create():
429425
sys.exit(1)
430426

431427
# Display the hardware instance information to the user.
432-
credits = selected_hw.cost_per_hr / 100.0 # e.g., 360 -> 3.60 credits per hour
433-
vram_gib = selected_hw.accelerator_memory / 1024 # e.g., 81920 MB -> 80 GiB VRAM
434-
memory_gib = selected_hw.memory / 1024 # e.g., 239616 MB -> 234 GiB memory
435-
cpu_cores = selected_hw.cpu / 1000 # e.g., 26000 millicores -> 26 cores
428+
credits = selected_hw.cost_per_hr / 100.0 # e.g., 360 -> 3.60 credits per hour
429+
vram_gib = selected_hw.accelerator_memory / 1024 # e.g., 81920 MB -> 80 GiB VRAM
430+
memory_gib = selected_hw.memory / 1024 # e.g., 239616 MB -> 234 GiB memory
431+
cpu_cores = selected_hw.cpu / 1000 # e.g., 26000 millicores -> 26 cores
436432

437433
click.echo("Selected Hardware Instance:")
438-
click.echo(f"{credits:.2f} credits per hour,\n{vram_gib:.0f}GiB VRAM,\nMemory {memory_gib:.0f}GiB,\nCPU {cpu_cores:.0f} cores")
434+
click.echo(f"{credits:.2f} credits per hour,\n{vram_gib:.0f}GiB VRAM,")
435+
click.echo(f"Memory {memory_gib:.0f}GiB,\nCPU {cpu_cores:.0f} cores")
439436

440437
# Use the cluster_id from the hardware instance (no need to prompt the user)
441438
cluster_id = selected_hw.cluster_id

centml/sdk/api.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,15 +59,18 @@ def resume(self, id):
5959
def get_clusters(self):
6060
return self._api.get_clusters_clusters_get()
6161

62-
def get_hardware_instances(self, cluster_id = None):
63-
return self._api.get_hardware_instances_hardware_instances_get(cluster_id = cluster_id if cluster_id else None).results
62+
def get_hardware_instances(self, cluster_id=None):
63+
return self._api.get_hardware_instances_hardware_instances_get(
64+
cluster_id=cluster_id if cluster_id else None
65+
).results
6466

6567
def get_prebuilt_images(self, depl_type: DeploymentType = None):
6668
return self._api.get_prebuilt_images_prebuilt_images_get(type=depl_type)
6769

6870
def get_cserve_recipe(self):
6971
return self._api.get_cserve_recipe_deployments_cserve_recipes_get().results
7072

73+
7174
@contextmanager
7275
def get_centml_client():
7376
configuration = platform_api_python_client.Configuration(

0 commit comments

Comments
 (0)