support old weights

ecmwf · Jun 16, 2024 · 41341db · 41341db
1 parent cea106e
commit 41341db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/anemoi/inference/checkpoint/metadata/__init__.py b/src/anemoi/inference/checkpoint/metadata/__init__.py
@@ -72,7 +72,7 @@ def from_metadata(cls, metadata):
 
         # When we changed from ecml_tools to anemoi-datasets, we went back in the
         # versionning
-        if metadata["dataset"]["version"] in ("0.1.7", "0.1.8", "0.1.9"):
+        if metadata["dataset"]["version"] in ("0.1.4", "0.1.7", "0.1.8", "0.1.9"):
             metadata["dataset"]["version"] = "0.2.0"
 
         klass = from_versions(metadata["version"], metadata["dataset"]["version"])