feat!: Implement coupling of two or more models

.gitignore

@@ -119,7 +119,6 @@ tmp/
 temp/
 logs/
 _dev/
-outputs/
 *tmp_data/
 
 # Project specific
@@ -135,3 +134,5 @@ _version.py
 *.to_upload
 tempCodeRunnerFile.python
 Untitled-*.py
+*.json
+*.txt

.pre-commit-config.yaml

@@ -78,7 +78,10 @@ repos:
         - --check-overridden # Check overridden methods
         - --check-protected  # Check protected methods
         - --check-class      # Check class docstrings
-        - --disable=SIG101,SIG102     # Disable empty docstrings
+        # - --disable=SIG101,SIG102     # Disable empty docstrings
+        # - --disable=SIG505,SIG503
+
+
 ci:
   autoupdate_schedule: monthly
   autoupdate_commit_msg: "chore(deps): pre-commit.ci autoupdate"

pyproject.toml

@@ -86,3 +86,12 @@ entry-points."ai_models.model".anemoi = "anemoi.inference.plugin:AIModelPlugin"
 
 [tool.setuptools_scm]
 version_file = "src/anemoi/inference/_version.py"
+
+[tool.mypy]
+exclude = [ "docs/" ]
+strict = false
+ignore_missing_imports = true
+allow_redefinition = true
+
+[setuptools.package-data]
+"anemoi.inference.grib.templates" = [ "*.yaml" ]

src/anemoi/inference/__init__.py

@@ -11,7 +11,9 @@
 try:
     # NOTE: the `_version.py` file must not be present in the git repository
     #   as it is generated by setuptools at install time
-    from ._version import __version__  # type: ignore
+    from ._version import __version__
 except ImportError:  # pragma: no cover
     # Local copy or not installed with setuptools
     __version__ = "999"
+
+__all__ = ["__version__"]

src/anemoi/inference/__main__.py

@@ -7,6 +7,8 @@
 # granted to it by virtue of its status as an intergovernmental organisation
 # nor does it submit to any jurisdiction.
 
+from typing import Any
+
 from anemoi.utils.cli import cli_main
 from anemoi.utils.cli import make_parser
 
@@ -15,11 +17,24 @@
 
 
 # For read-the-docs
-def create_parser():
+def create_parser() -> Any:
+    """Create a command-line argument parser.
+
+    Returns
+    -------
+    Any
+        The command-line argument parser.
+    """
     return make_parser(__doc__, COMMANDS)
 
 
-def main():
+def main() -> None:
+    """Execute the main command-line interface.
+
+    Returns
+    -------
+    None
+    """
     cli_main(__version__, __doc__, COMMANDS)