Skip to content

Commit

Permalink
Rename job.py to fm_step.py
Browse files Browse the repository at this point in the history
  • Loading branch information
berland committed Oct 23, 2024
1 parent e55b1f9 commit ec730d9
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 8 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,7 @@ def killed_by_oom(pids: Sequence[int]) -> bool:
return False


class Job:
class FMstep:
MEMORY_POLL_PERIOD = 5 # Seconds between memory polls

def __init__(
Expand Down
8 changes: 4 additions & 4 deletions src/_ert/forward_model_runner/reporting/message.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
import psutil

if TYPE_CHECKING:
from _ert.forward_model_runner.job import Job
from _ert.forward_model_runner.job import FMstep

class _ChecksumDictBase(TypedDict):
type: Literal["file"]
Expand Down Expand Up @@ -71,7 +71,7 @@ def __repr__(cls):
class Message(metaclass=_MetaMessage):
def __init__(self, job=None):
self.timestamp = dt.now()
self.job: Optional[Job] = job
self.job: Optional[FMstep] = job
self.error_message: Optional[str] = None

def __repr__(self):
Expand Down Expand Up @@ -116,12 +116,12 @@ def __init__(self):


class Start(Message):
def __init__(self, job: "Job"):
def __init__(self, job: "FMstep"):
super().__init__(job)


class Running(Message):
def __init__(self, job: "Job", memory_status: ProcessTreeStatus):
def __init__(self, job: "FMstep", memory_status: ProcessTreeStatus):
super().__init__(job)
self.memory_status = memory_status

Expand Down
6 changes: 3 additions & 3 deletions src/_ert/forward_model_runner/runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from pathlib import Path
from typing import List

from _ert.forward_model_runner.job import Job
from _ert.forward_model_runner.job import FMstep
from _ert.forward_model_runner.reporting.message import Checksum, Finish, Init


Expand All @@ -21,9 +21,9 @@ def __init__(self, jobs_data):
if self.simulation_id is not None:
os.environ["ERT_RUN_ID"] = self.simulation_id

self.jobs: List[Job] = []
self.jobs: List[FMstep] = []
for index, job_data in enumerate(job_data_list):
self.jobs.append(Job(job_data, index))
self.jobs.append(FMstep(job_data, index))

self._set_environment()

Expand Down

0 comments on commit ec730d9

Please sign in to comment.