Make max_submit parameter function instead

xjules · Dec 13, 2023 · 8364762 · 8364762
1 parent 3e725e8
commit 8364762
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/src/ert/scheduler/job.py b/src/ert/scheduler/job.py
@@ -66,7 +66,7 @@ def driver(self) -> Driver:
         return self._scheduler.driver
 
     async def __call__(
-        self, start: asyncio.Event, sem: asyncio.BoundedSemaphore
+        self, start: asyncio.Event, sem: asyncio.BoundedSemaphore, max_submit: int = 2
     ) -> None:
         await start.wait()
         retries = 0
@@ -97,18 +97,16 @@ async def __call__(
                 else:
                     await self._send(State.FAILED)
                     retries += 1
-                    retry = retries < self._scheduler._max_submit
+                    retry = retries < max_submit
                     if retry:
                         message = f"Realization: {self.iens} failed, resubmitting"
                         logger.warning(message)
-                        print(message)
                     else:
                         message = (
                             f"Realization: {self.iens} "
                             "failed after reaching max submit "
                             f"{self._scheduler._max_submit}!"
                         )
-                        print(message)
                         logger.error(message)
 
             except asyncio.CancelledError:

diff --git a/src/ert/scheduler/scheduler.py b/src/ert/scheduler/scheduler.py
@@ -44,6 +44,7 @@ def __init__(self, driver: Optional[Driver] = None) -> None:
         self._tasks: MutableMapping[int, asyncio.Task[None]] = {}
 
         self._events: Optional[asyncio.Queue[Any]] = None
+        # will be read from QueueConfig
         self._max_submit: int = 2
 
         self._ee_uri = ""
@@ -121,7 +122,7 @@ async def execute(
         start = asyncio.Event()
         sem = asyncio.BoundedSemaphore(semaphore._initial_value if semaphore else 10)  # type: ignore
         for iens, job in self._jobs.items():
-            self._tasks[iens] = asyncio.create_task(job(start, sem))
+            self._tasks[iens] = asyncio.create_task(job(start, sem, self._max_submit))
 
         start.set()
         for task in self._tasks.values():