ert 18.0.9__py3-none-any.whl → 19.0.0__py3-none-any.whl

ert/storage/migration/to20.py

@@ -0,0 +1,23 @@
+import json
+from pathlib import Path
+from typing import Any
+
+info = "Remove redundant .name attribute from responses."
+
+
+def config_without_name_attr(config: dict[str, Any]) -> dict[str, Any]:
+    new_json = {**config}
+    new_json.pop("name", None)
+
+    return new_json
+
+
+def migrate(path: Path) -> None:
+    for response_json_path in path.glob("experiments/*/responses.json"):
+        old_json = json.loads((response_json_path).read_text(encoding="utf-8"))
+        new_json = {
+            response_type: config_without_name_attr(config)
+            for response_type, config in old_json.items()
+        }
+
+        response_json_path.write_text(json.dumps(new_json, indent=2), encoding="utf-8")

ert/storage/migration/to21.py

@@ -0,0 +1,25 @@
+import json
+from pathlib import Path
+from typing import Any
+
+info = "Remove refcase from summary response configs"
+
+
+def config_without_refcase(summary_config: dict[str, Any]) -> dict[str, Any]:
+    new_json = {**summary_config}
+    new_json.pop("refcase", None)
+
+    return new_json
+
+
+def migrate(path: Path) -> None:
+    for response_json_path in path.glob("experiments/*/responses.json"):
+        old_json = json.loads((response_json_path).read_text(encoding="utf-8"))
+        new_json = {
+            response_type: config_without_refcase(response_config)
+            if response_config["type"] == "summary"
+            else response_config
+            for response_type, response_config in old_json.items()
+        }
+
+        response_json_path.write_text(json.dumps(new_json, indent=2), encoding="utf-8")

ert/storage/migration/to22.py

@@ -0,0 +1,18 @@
+from pathlib import Path
+
+import polars as pl
+
+info = "Add default None values to summary observations LOCATION keywords"
+
+
+def migrate(path: Path) -> None:
+    for summary_observation in path.glob("experiments/*/observations/summary"):
+        summary_df = pl.read_parquet(summary_observation)
+
+        for location_kw in ["location_x", "location_y", "location_range"]:
+            if location_kw not in summary_df.columns:
+                summary_df = summary_df.with_columns(
+                    pl.lit(None, dtype=pl.Float32).alias(location_kw)
+                )
+
+        summary_df.write_parquet(summary_observation)

ert/storage/migration/to23.py

@@ -0,0 +1,49 @@
+from pathlib import Path
+
+import polars as pl
+
+info = "Add default None values to summary observations LOCATION keywords"
+
+old_localization_keywords = ["location_x", "location_y", "location_range"]
+new_localization_keywords = ["east", "north", "radius"]
+
+
+def migrate(path: Path) -> None:
+    for gen_obs in path.glob("experiments/*/observations/gen_data"):
+        gen_obs_df = pl.read_parquet(gen_obs)
+
+        for new_kw in new_localization_keywords:
+            if new_kw not in gen_obs_df.columns:
+                gen_obs_df = gen_obs_df.with_columns(
+                    pl.lit(None, dtype=pl.Float32).alias(new_kw)
+                )
+
+        gen_obs_df.write_parquet(gen_obs)
+
+    for rft_obs in path.glob("experiments/*/observations/rft"):
+        rft_obs_df = pl.read_parquet(rft_obs)
+
+        for new_kw in new_localization_keywords:
+            if new_kw not in rft_obs_df.columns:
+                rft_obs_df = rft_obs_df.with_columns(
+                    pl.lit(None, dtype=pl.Float32).alias(new_kw)
+                )
+
+        rft_obs_df.write_parquet(rft_obs)
+
+    for summary_obs in path.glob("experiments/*/observations/summary"):
+        summary_df = pl.read_parquet(summary_obs)
+
+        for old_kw, new_kw in zip(
+            old_localization_keywords, new_localization_keywords, strict=True
+        ):
+            if old_kw in summary_df.columns:
+                column = summary_df[old_kw]
+                summary_df = summary_df.with_columns(column.alias(new_kw))
+                summary_df = summary_df.drop(old_kw)
+            else:
+                summary_df = summary_df.with_columns(
+                    pl.lit(None, dtype=pl.Float32).alias(new_kw)
+                )
+
+        summary_df.write_parquet(summary_obs)

ert/workflow_runner.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import logging
+import types
 from concurrent import futures
 from concurrent.futures import Future
 from typing import Any, Self
@@ -127,7 +128,7 @@ class WorkflowRunner:
         self,
         exc_type: type[BaseException] | None,
         exc_value: BaseException | None,
-        traceback: Any,
+        traceback: types.TracebackType | None,
     ) -> None:
         self.wait()
 

{ert-18.0.9.dist-info → ert-19.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ert
-Version: 18.0.9
+Version: 19.0.0
 Summary: Ensemble based Reservoir Tool (ERT)
 Author-email: Equinor ASA <fg_sib-scout@equinor.com>
 License-Expression: GPL-3.0-only