@@ -149,7 +149,7 @@ async def create_a_vm(
149
149
try :
150
150
if message .requirements and message .requirements .gpu :
151
151
# Ensure we have the necessary GPU for the user by reserving them
152
- resources = self .find_resources_available_for_user (cast ( message , InstanceContent ) , message .address )
152
+ resources = self .find_resources_available_for_user (message , message .address )
153
153
# First assign Host GPUs from the available
154
154
execution .prepare_gpus (list (resources ))
155
155
# Prepare VM general Resources and also the GPUs
@@ -383,7 +383,7 @@ def get_valid_reservation(self, resource) -> Reservation | None:
383
383
del self .reservations [resource ]
384
384
return self .reservations .get (resource )
385
385
386
- async def reserve_resources (self , message : InstanceContent , user ):
386
+ async def reserve_resources (self , message : ExecutableContent , user ):
387
387
gpu_to_reserve = message .requirements .gpu if message .requirements and message .requirements .gpu else []
388
388
expiration_date = datetime .now (tz = timezone .utc ) + timedelta (seconds = 60 )
389
389
if not gpu_to_reserve :
@@ -400,8 +400,8 @@ async def reserve_resources(self, message: InstanceContent, user):
400
400
401
401
return expiration_date
402
402
403
- def find_resources_available_for_user (self , message : InstanceContent , user ) -> set [GpuDevice ]:
404
- """Find required resource to run InstanceContent from reserved resources by user or free resources.
403
+ def find_resources_available_for_user (self , message : ExecutableContent , user ) -> set [GpuDevice ]:
404
+ """Find required resource to run ExecutableContent from reserved resources by user or free resources.
405
405
406
406
Only implement GPU for now"""
407
407
# Calling function should use the creation_lock to avoid resource being stollem
0 commit comments