improve typing for storagemodel

src/ert/gui/tools/manage_experiments/storage_model.py

@@ -26,7 +26,7 @@ class _Column(IntEnum):
 
 
 class RealizationModel:
-    def __init__(self, realization: int, parent: Any) -> None:
+    def __init__(self, realization: int, parent: "EnsembleModel") -> None:
         self._parent = parent
         self._name = f"Realization {realization}"
         self._ensemble_id = parent._id
@@ -58,7 +58,7 @@ def data(self, index: QModelIndex, role: Qt.ItemDataRole) -> Any:
 
 
 class EnsembleModel:
-    def __init__(self, ensemble: Ensemble, parent: Any):
+    def __init__(self, ensemble: Ensemble, parent: "ExperimentModel"):
         self._parent = parent
         self._name = ensemble.name
         self._id = ensemble.id
@@ -191,9 +191,8 @@ def parent(self, child: QModelIndex | None = None) -> QObject | QModelIndex:
         child_item = cast(ChildModel, child.internalPointer())
         parentItem = child_item._parent
 
-        if parentItem == self:
+        if isinstance(parentItem, StorageModel):
             return QModelIndex()
-
         return self.createIndex(parentItem.row(), 0, parentItem)
 
     @override