Skip to content

Commit 2b63b6d

Browse files
lindjohaØyvind Lind-Johansen
andauthored
Expose rft input files as arguments (#1313)
* exposed rft input files as arguments * black version updated * pylint fixes * pylint fixes * more fixes --------- Co-authored-by: Øyvind Lind-Johansen <[email protected]>
1 parent 6d1b6f6 commit 2b63b6d

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

webviz_subsurface/plugins/_rft_plotter/_plugin.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -102,18 +102,23 @@ class Ids(StrEnum):
102102
def __init__(
103103
self,
104104
webviz_settings: WebvizSettings,
105+
rft: str = "share/results/tables/rft.csv",
106+
rft_ert: str = "share/results/tables/rft_ert.csv",
105107
csvfile_rft: Path = None,
106108
csvfile_rft_ert: Path = None,
107109
ensembles: Optional[List[str]] = None,
108110
formations: Path = None,
109111
obsdata: Path = None,
110112
faultlines: Path = None,
111113
) -> None:
114+
# pylint: disable = too-many-arguments
112115
super().__init__()
113116

114117
self._datamodel = RftPlotterDataModel(
115118
webviz_settings,
116119
ensembles,
120+
rft,
121+
rft_ert,
117122
formations,
118123
faultlines,
119124
obsdata,

webviz_subsurface/plugins/_rft_plotter/_utils/_rft_plotter_data_model.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,12 +27,15 @@ def __init__(
2727
self,
2828
webviz_settings: WebvizSettings,
2929
ensembles: Optional[List[str]],
30+
rft: str,
31+
rft_ert: str,
3032
formations: Path = None,
3133
faultlines: Path = None,
3234
obsdata: Path = None,
3335
csvfile_rft: Path = None,
3436
csvfile_rft_ert: Path = None,
3537
):
38+
# pylint: disable = too-many-arguments
3639
self.formations = formations
3740
self.faultlines = faultlines
3841
self.obsdata = obsdata
@@ -67,12 +70,12 @@ def __init__(
6770

6871
self.ertdatadf = create_csvfile_providerset_from_paths(
6972
ens_paths,
70-
"share/results/tables/rft_ert.csv",
73+
rft_ert,
7174
).get_aggregated_dataframe()
7275

7376
try:
7477
self.simdf = create_csvfile_providerset_from_paths(
75-
ens_paths, "share/results/tables/rft.csv"
78+
ens_paths, rft
7679
).get_aggregated_dataframe()
7780

7881
except ValueError as err:

0 commit comments

Comments
 (0)