fix can_run signature

ecmwf-projects · Nov 20, 2024 · b566799 · b566799
1 parent 5d20448
commit b566799
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cads_broker/dispatcher.py b/cads_broker/dispatcher.py
@@ -656,7 +656,7 @@ def submit_requests(
         requests_counter = 0
         for request in queue:
             if self.qos.can_run(
-                request, session=session_write, scheduler=self.internal_scheduler
+                request, scheduler=self.internal_scheduler
             ):
                 if requests_counter <= int(number_of_requests):
                     self.submit_request(request, session=session_write)

diff --git a/cads_broker/qos/QoS.py b/cads_broker/qos/QoS.py
@@ -290,7 +290,7 @@ def user_limit(self, request):
     @locked
     def pick(self, queue, session):
         # Create the list of requests than can run
-        candidates = [(n, r) for n, r in enumerate(queue) if self.can_run(r, session)]
+        candidates = [(n, r) for n, r in enumerate(queue) if self.can_run(r)]
 
         # If no request can run, return 'None'
         if len(candidates) == 0: