fix: config import

src/anemoi/models/interface/__init__.py

@@ -10,7 +10,7 @@
 import uuid
 
 import torch
-from anemoi.utils.config import DotConfig
+from anemoi.utils.config import DotDict
 from hydra.utils import instantiate
 from torch_geometric.data import HeteroData
 
@@ -19,10 +19,37 @@
 
 
 class AnemoiModelInterface(torch.nn.Module):
-    """Anemoi model on torch level."""
+    """An interface for Anemoi models.
+
+    This class is a wrapper around the Anemoi model that includes pre-processing and post-processing steps.
+    It inherits from the PyTorch Module class.
+
+    Attributes
+    ----------
+    config : DotConfig
+        Configuration settings for the model.
+    id : str
+        A unique identifier for the model instance.
+    multi_step : bool
+        Whether the model uses multi-step input.
+    graph_data : HeteroData
+        Graph data for the model.
+    statistics : dict
+        Statistics for the data.
+    metadata : dict
+        Metadata for the model.
+    data_indices : dict
+        Indices for the data.
+    pre_processors : Processors
+        Pre-processing steps to apply to the data before passing it to the model.
+    post_processors : Processors
+        Post-processing steps to apply to the model's output.
+    model : AnemoiModelEncProcDec
+        The underlying Anemoi model.
+    """
 
     def __init__(
-        self, *, config: DotConfig, graph_data: HeteroData, statistics: dict, data_indices: dict, metadata: dict
+        self, *, config: DotDict, graph_data: HeteroData, statistics: dict, data_indices: dict, metadata: dict
     ) -> None:
         super().__init__()
         self.config = config
@@ -35,7 +62,7 @@ def __init__(
         self._build_model()
 
     def _build_model(self) -> None:
-        """Build the model and pre- and post-processors."""
+        """Builds the model and pre- and post-processors."""
         # Instantiate processors
         processors = [
             [name, instantiate(processor, statistics=self.statistics, data_indices=self.data_indices)]