Skip to content

Commit c4bc4e1

Browse files
committed
review fixes
1 parent 60f6d8c commit c4bc4e1

2 files changed

Lines changed: 11 additions & 10 deletions

File tree

src/aleph/sdk/client/services/crn.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ class DiskInfo(BaseModel):
5757

5858
class UsagePeriod(BaseModel):
5959
start_timestamp: datetime
60-
duration_seconds: int
60+
duration_seconds: NonNegativeInt
6161

6262

6363
class Properties(BaseModel):
@@ -206,24 +206,24 @@ def filter_crn(
206206

207207
# Filter VM resources
208208
if vm_resources:
209-
sys = crn.system_usage
210-
if not sys:
209+
crn_usage = crn.system_usage
210+
if not crn_usage:
211211
continue
212212

213213
# Check CPU count
214-
if sys.cpu.count < vm_resources.vcpus:
214+
if crn_usage.cpu.count < vm_resources.vcpus:
215215
continue
216216

217217
# Convert MiB to kB (1 MiB = 1024 kB) for proper comparison
218-
memory_kb_required = vm_resources.memory * 1024
218+
memory_kb_required = vm_resources.memory_mib * 1024
219219
disk_kb_required = vm_resources.disk_mib * 1024
220220

221221
# Check free memory
222-
if sys.mem.available_kB < memory_kb_required:
222+
if crn_usage.mem.available_kB < memory_kb_required:
223223
continue
224224

225225
# Check free disk
226-
if sys.disk.available_kB < disk_kb_required:
226+
if crn_usage.disk.available_kB < disk_kb_required:
227227
continue
228228

229229
filtered_crn.append(crn)

src/aleph/sdk/types.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
BaseModel,
2020
ConfigDict,
2121
Field,
22+
PositiveInt,
2223
RootModel,
2324
TypeAdapter,
2425
field_validator,
@@ -402,6 +403,6 @@ class Voucher(BaseModel):
402403

403404

404405
class VmResources(BaseModel):
405-
vcpus: int
406-
memory: int
407-
disk_mib: int
406+
vcpus: PositiveInt
407+
memory_mib: PositiveInt
408+
disk_mib: PositiveInt

0 commit comments

Comments
 (0)