Handle ensemble schema without parent and child ensemble_ids

equinor · Aug 12, 2024 · 5b8b6cc · 5b8b6cc
1 parent 25e6e28
commit 5b8b6cc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/tests/models/test_ensemble_model.py b/tests/models/test_ensemble_model.py
@@ -7,8 +7,6 @@
 
 def test_ensemble_model(mock_data):
     ens_model = EnsembleModel(ensemble_id=1, project_id=None)
-    assert len(ens_model.children) == 1
-    assert ens_model.children[0].name == "default_smoother_update"
     assert ens_model.name == "default"
     assert len(ens_model.responses) == 1
 

diff --git a/webviz_ert/models/ensemble_model.py b/webviz_ert/models/ensemble_model.py
@@ -43,8 +43,8 @@ def __init__(self, ensemble_id: str, project_id: str) -> None:
         self._metadata = self._schema["userdata"]
         self._name = self._metadata["name"]
         self._id = ensemble_id
-        self._children = self._schema["child_ensemble_ids"]
-        self._parent = self._schema["parent_ensemble_id"]
+        self._children = self._schema.get("child_ensemble_ids", [])
+        self._parent = self._schema.get("parent_ensemble_id", [])
         self._size = self._schema["size"]
         self._active_realizations = self._schema["active_realizations"]
         self._responses: Dict[str, Response] = {}