Propagate exec_hosts to scheduler

Author: andreas-el

src/ert/scheduler/event.py

@@ -7,12 +7,14 @@
 @dataclass
 class StartedEvent:
     iens: int
+    exec_hosts: str
 
 
 @dataclass
 class FinishedEvent:
     iens: int
     returncode: int
+    exec_hosts: str
 
 
 Event = Union[StartedEvent, FinishedEvent]

src/ert/scheduler/lsf_driver.py

@@ -503,12 +503,15 @@ async def _process_job_update(self, job_id: str, new_state: AnyJob) -> None:
         event: Optional[Event] = None
         if isinstance(new_state, RunningJob):
             logger.debug(f"Realization {iens} is running")
-            event = StartedEvent(iens=iens)
+            event = StartedEvent(iens=iens, exec_hosts=self._jobs[job_id].exec_hosts)
         elif isinstance(new_state, FinishedJobFailure):
             logger.info(f"Realization {iens} (LSF-id: {self._iens2jobid[iens]}) failed")
             exit_code = await self._get_exit_code(job_id)
-            event = FinishedEvent(iens=iens, returncode=exit_code)
-
+            event = FinishedEvent(
+                iens=iens,
+                returncode=exit_code,
+                exec_hosts=self._jobs[job_id].exec_hosts,
+            )
         elif isinstance(new_state, FinishedJobSuccess):
             logger.info(
                 f"Realization {iens} (LSF-id: {self._iens2jobid[iens]}) succeeded"

src/ert/scheduler/scheduler.py

@@ -28,7 +28,7 @@
 from ert.constant_filenames import CERT_FILE
 
 from .driver import Driver
-from .event import FinishedEvent
+from .event import FinishedEvent, StartedEvent
 from .job import Job, JobState
 
 if TYPE_CHECKING:
@@ -308,6 +308,9 @@ async def _process_event_queue(self) -> None:
             # Any event implies the job has at least started
             job.started.set()
 
+            if isinstance(event, (StartedEvent, FinishedEvent)):
+                logger.warning(f"lsf exec_hosts: {event.exec_hosts}")
+
             if (
                 isinstance(event, FinishedEvent)
                 and not self._cancelled