Skip to content

Commit 947cea8

Browse files
authored
Show info when an instance is spot compute in availability overview (#19)
* Show info when an instance is spot compute
1 parent 2346019 commit 947cea8

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[project]
22
name = "prime-cli"
3-
version = "0.2.2"
3+
version = "0.2.3"
44
description = "Prime Intellect CLI"
55
readme = "README.md"
66
requires-python = ">=3.8"

src/prime_cli/commands/availability.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,7 @@ def list(
118118
"security": gpu.security or "N/A",
119119
"vcpu": gpu.vcpu.default_count,
120120
"memory": gpu.memory.default_count,
121+
"is_spot": gpu.is_spot,
121122
}
122123
all_gpus.append(gpu_data)
123124

@@ -163,9 +164,14 @@ def list(
163164
filtered_gpus.append(gpu_config)
164165

165166
for gpu_entry in filtered_gpus:
167+
gpu_type_display = (
168+
f"{gpu_entry['gpu_type']} (Spot)"
169+
if gpu_entry["is_spot"]
170+
else gpu_entry["gpu_type"]
171+
)
166172
table.add_row(
167173
gpu_entry["short_id"],
168-
gpu_entry["gpu_type"],
174+
gpu_type_display,
169175
str(gpu_entry["gpu_count"]),
170176
gpu_entry["socket"],
171177
gpu_entry["provider"],

0 commit comments

Comments
 (0)