Skip to content

Commit

Permalink
Rename ext_job* to forward_model_*
Browse files Browse the repository at this point in the history
  • Loading branch information
berland committed Nov 7, 2023
1 parent ca92d52 commit 15c4ae4
Show file tree
Hide file tree
Showing 9 changed files with 151 additions and 131 deletions.
8 changes: 4 additions & 4 deletions src/ert/config/forward_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,9 +9,9 @@
from .parse_arg_types_list import parse_arg_types_list
from .parsing import (
ConfigValidationError,
ExtJobKeys,
ForwardModelKeys,
SchemaItemType,
init_ext_job_schema,
init_forward_model_schema,
lark_parse,
)

Expand Down Expand Up @@ -54,13 +54,13 @@ def from_config_file(
if name is None:
name = os.path.basename(config_file)

schema = init_ext_job_schema()
schema = init_forward_model_schema()

try:
content_dict = lark_parse(file=config_file, schema=schema, pre_defines=[])

specified_arg_types: List[Tuple[int, str]] = content_dict.get(
ExtJobKeys.ARG_TYPE, []
ForwardModelKeys.ARG_TYPE, []
)

specified_max_args: int = content_dict.get("MAX_ARG", 0)
Expand Down
8 changes: 4 additions & 4 deletions src/ert/config/parsing/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,8 @@
from .config_schema import init_site_config_schema, init_user_config_schema
from .context_values import ContextList, ContextString, ContextValue
from .error_info import ErrorInfo, WarningInfo
from .ext_job_keywords import ExtJobKeys
from .ext_job_schema import init_ext_job_schema
from .forward_model_keywords import ForwardModelKeys
from .forward_model_schema import init_forward_model_schema
from .lark_parser import parse as lark_parse
from .schema_item_type import SchemaItemType
from .types import MaybeWithContext
Expand All @@ -25,10 +25,10 @@
"init_site_config_schema",
"init_user_config_schema",
"ConfigDict",
"ExtJobKeys",
"ForwardModelKeys",
"ErrorInfo",
"WarningInfo",
"init_ext_job_schema",
"init_forward_model_schema",
"ContextString",
"MaybeWithContext",
"ContextList",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ class StrEnum(str, Enum):
from enum import StrEnum


class ExtJobKeys(StrEnum):
class ForwardModelKeys(StrEnum):
NAME = "NAME"
EXECUTABLE = "EXECUTABLE"

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,104 +3,106 @@
from .config_dict import ConfigDict
from .config_schema_item import SchemaItem
from .deprecation_info import DeprecationInfo
from .ext_job_keywords import ExtJobKeys
from .forward_model_keywords import ForwardModelKeys
from .schema_dict import SchemaItemDict
from .schema_item_type import SchemaItemType


def executable_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.EXECUTABLE,
kw=ForwardModelKeys.EXECUTABLE,
required_set=True,
type_map=[SchemaItemType.EXECUTABLE],
)


def stdin_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.STDIN,
kw=ForwardModelKeys.STDIN,
type_map=[SchemaItemType.STRING],
required_set=False,
)


def stdout_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.STDOUT,
kw=ForwardModelKeys.STDOUT,
type_map=[SchemaItemType.STRING],
required_set=False,
)


def stderr_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.STDERR,
kw=ForwardModelKeys.STDERR,
type_map=[SchemaItemType.STRING],
required_set=False,
)


def start_file_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.START_FILE,
kw=ForwardModelKeys.START_FILE,
type_map=[SchemaItemType.STRING],
)


def target_file_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.TARGET_FILE,
kw=ForwardModelKeys.TARGET_FILE,
type_map=[SchemaItemType.STRING],
)


def error_file_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.ERROR_FILE,
kw=ForwardModelKeys.ERROR_FILE,
type_map=[SchemaItemType.STRING],
)


def max_running_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.MAX_RUNNING, type_map=[SchemaItemType.INT], required_set=False
kw=ForwardModelKeys.MAX_RUNNING,
type_map=[SchemaItemType.INT],
required_set=False,
)


def max_running_minutes_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.MAX_RUNNING_MINUTES,
kw=ForwardModelKeys.MAX_RUNNING_MINUTES,
type_map=[SchemaItemType.INT],
required_set=False,
)


def min_arg_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.MIN_ARG,
kw=ForwardModelKeys.MIN_ARG,
type_map=[SchemaItemType.INT],
required_set=False,
)


def max_arg_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.MAX_ARG,
kw=ForwardModelKeys.MAX_ARG,
type_map=[SchemaItemType.INT],
required_set=False,
)


def arglist_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.ARGLIST,
kw=ForwardModelKeys.ARGLIST,
argc_max=None,
)


def arg_type_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.ARG_TYPE,
kw=ForwardModelKeys.ARG_TYPE,
argc_min=2,
argc_max=2,
type_map=[SchemaItemType.INT, SchemaItemType.STRING],
Expand All @@ -110,7 +112,7 @@ def arg_type_keyword() -> SchemaItem:

def env_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.ENV,
kw=ForwardModelKeys.ENV,
argc_min=2,
argc_max=2,
multi_occurrence=True,
Expand All @@ -120,7 +122,7 @@ def env_keyword() -> SchemaItem:

def exec_env_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.EXEC_ENV,
kw=ForwardModelKeys.EXEC_ENV,
argc_min=2,
argc_max=2,
multi_occurrence=True,
Expand All @@ -130,15 +132,15 @@ def exec_env_keyword() -> SchemaItem:

def default_keyword() -> SchemaItem:
return SchemaItem(
kw=ExtJobKeys.DEFAULT,
kw=ForwardModelKeys.DEFAULT,
argc_min=2,
argc_max=2,
multi_occurrence=True,
type_map=[SchemaItemType.STRING, SchemaItemType.STRING],
)


ext_job_schema_items: List[SchemaItem] = [
forward_model_schema_items: List[SchemaItem] = [
executable_keyword(),
stdin_keyword(),
stdout_keyword(),
Expand All @@ -157,15 +159,15 @@ def default_keyword() -> SchemaItem:
exec_env_keyword(),
]

ext_job_deprecations: List[DeprecationInfo] = [
forward_model_deprecations: List[DeprecationInfo] = [
DeprecationInfo(
keyword="PORTABLE_EXE",
message='"PORTABLE_EXE" key is deprecated, please replace with "EXECUTABLE"',
)
]


class ExtJobSchemaItemDict(SchemaItemDict):
class ForwardModelSchemaItemDict(SchemaItemDict):
def check_required(
self,
config_dict: ConfigDict,
Expand All @@ -180,11 +182,11 @@ def check_required(
)


def init_ext_job_schema() -> ExtJobSchemaItemDict:
schema = ExtJobSchemaItemDict()
def init_forward_model_schema() -> ForwardModelSchemaItemDict:
schema = ForwardModelSchemaItemDict()

for item in ext_job_schema_items:
for item in forward_model_schema_items:
schema[item.kw] = item

schema.add_deprecations(ext_job_deprecations)
schema.add_deprecations(forward_model_deprecations)
return schema
6 changes: 3 additions & 3 deletions tests/unit_tests/config/config_dict_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -167,7 +167,7 @@ def queue_options(draw, systems):
return [queue_system, name]


def default_ext_job_names():
def default_forward_model_names():
return (
st.sampled_from(
[
Expand Down Expand Up @@ -207,7 +207,7 @@ def default_ext_job_names():


@st.composite
def random_ext_job_names(draw, some_words, some_file_names):
def random_forward_model_names(draw, some_words, some_file_names):
return draw(
st.tuples(
some_words,
Expand Down Expand Up @@ -387,7 +387,7 @@ def composite_keys(smspec: Smspec) -> st.SearchStrategy[str]:
def ert_config_values(draw, use_eclbase=booleans):
use_eclbase = draw(use_eclbase)
queue_system = draw(queue_systems)
install_jobs = draw(small_list(random_ext_job_names(words, file_names)))
install_jobs = draw(small_list(random_forward_model_names(words, file_names)))
forward_model = draw(small_list(job(install_jobs))) if install_jobs else []
simulation_job = draw(small_list(sim_job(install_jobs))) if install_jobs else []
gen_data = draw(
Expand Down
Loading

0 comments on commit 15c4ae4

Please sign in to comment.