Add defaults to validate_environment (#68)

* Add defaults * Update deps
ecmwf · Dec 3, 2024 · 97fa2a5 · 97fa2a5
1 parent 06292f9
commit 97fa2a5
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -41,7 +41,7 @@ classifiers = [
 dynamic = [ "version" ]
 dependencies = [
   "anemoi-transform>=0.1",
-  "anemoi-utils[text]>=0.4.9",
+  "anemoi-utils[text,provenance]>=0.4.9",
   "aniso8601",
   "anytree",
   "earthkit-data>=0.11.2",

diff --git a/src/anemoi/inference/checkpoint.py b/src/anemoi/inference/checkpoint.py
@@ -124,7 +124,13 @@ def default_namer(self, *args, **kwargs):
     def report_error(self):
         self._metadata.report_error()
 
-    def validate_environment(self, *, all_packages, on_difference, exempt_packages):
+    def validate_environment(
+        self,
+        *,
+        all_packages: bool = False,
+        on_difference: str = "warn",
+        exempt_packages: list[str] | None = None,
+    ) -> bool:
         return self._metadata.validate_environment(
             all_packages=all_packages, on_difference=on_difference, exempt_packages=exempt_packages
         )

diff --git a/src/anemoi/inference/provenance.py b/src/anemoi/inference/provenance.py
@@ -39,7 +39,7 @@
 
 
 def validate_environment(
-    metadata: Metadata,
+    metadata: "Metadata",
     *,
     all_packages: bool = False,
     on_difference: Literal["warn", "error", "ignore"] = "warn",