exposed rft input files as arguments

equinor · Jan 13, 2025 · a294de3 · a294de3
1 parent a8fdc13
commit a294de3
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/webviz_subsurface/plugins/_rft_plotter/_plugin.py b/webviz_subsurface/plugins/_rft_plotter/_plugin.py
@@ -102,6 +102,8 @@ class Ids(StrEnum):
     def __init__(
         self,
         webviz_settings: WebvizSettings,
+        rft: Path = "share/results/tables/rft.csv",
+        rft_ert: Path = "share/results/tables/rft_ert.csv",
         csvfile_rft: Path = None,
         csvfile_rft_ert: Path = None,
         ensembles: Optional[List[str]] = None,

diff --git a/webviz_subsurface/plugins/_rft_plotter/_utils/_rft_plotter_data_model.py b/webviz_subsurface/plugins/_rft_plotter/_utils/_rft_plotter_data_model.py
@@ -27,6 +27,8 @@ def __init__(
         self,
         webviz_settings: WebvizSettings,
         ensembles: Optional[List[str]],
+        rft: Path = None,
+        rft_ert: Path = None,
         formations: Path = None,
         faultlines: Path = None,
         obsdata: Path = None,
@@ -67,12 +69,12 @@ def __init__(
 
             self.ertdatadf = create_csvfile_providerset_from_paths(
                 ens_paths,
-                "share/results/tables/rft_ert.csv",
+                rft_ert,
             ).get_aggregated_dataframe()
 
             try:
                 self.simdf = create_csvfile_providerset_from_paths(
-                    ens_paths, "share/results/tables/rft.csv"
+                    ens_paths, rft
                 ).get_aggregated_dataframe()
 
             except ValueError as err:
@@ -210,7 +212,13 @@ def create_rft_and_param_pivot_table(
         zone: str,
         reals: List[int],
         keep_all_rfts: bool,
-    ) -> Tuple[Optional[pd.DataFrame], float, float, List[str], List[str],]:
+    ) -> Tuple[
+        Optional[pd.DataFrame],
+        float,
+        float,
+        List[str],
+        List[str],
+    ]:
         """This method merges rft observations and parameters.
 
         The RFT observations are converted from long form (all simulated values