@@ -43,7 +43,7 @@ async def add_resources(self, resources: list[str]):
4343 self .resources [resource ] = None
4444 if len (self .waiters ) > 0 :
4545 next_holder = self .waiters .pop (0 )
46- await self .allocate_resource (resource , next_holder )
46+ await self .assign_resource (resource , next_holder )
4747
4848 @workflow .signal
4949 async def acquire_resource (self , request : AcquireRequest ):
@@ -54,7 +54,7 @@ async def acquire_resource(self, request: AcquireRequest):
5454 for resource , holder in self .resources .items ():
5555 # Naively give out the first free resource, if we have one
5656 if holder is None :
57- await self .allocate_resource (resource , internal_request )
57+ await self .assign_resource (resource , internal_request )
5858 return
5959
6060 # Otherwise queue the request
@@ -63,7 +63,7 @@ async def acquire_resource(self, request: AcquireRequest):
6363 f"workflow_id={ request .workflow_id } is waiting for a resource"
6464 )
6565
66- async def allocate_resource (
66+ async def assign_resource (
6767 self , resource : str , internal_request : InternalAcquireRequest
6868 ):
6969 self .resources [resource ] = internal_request
@@ -108,7 +108,7 @@ async def release_resource(self, signal_name, *args):
108108 # If there are queued requests, assign the resource to the next one
109109 if len (self .waiters ) > 0 :
110110 next_holder = self .waiters .pop (0 )
111- await self .allocate_resource (resource , next_holder )
111+ await self .assign_resource (resource , next_holder )
112112
113113 @workflow .query
114114 def get_current_holders (self ) -> dict [str , Optional [InternalAcquireRequest ]]:
0 commit comments