add plugin

src/anemoi/inference/plugin.py

@@ -12,28 +12,34 @@
 import tqdm
 from ai_models.model import Model
 
+from anemoi.inference.runner import AUTOCAST
 from anemoi.inference.runner import DefaultRunner
 
 LOG = logging.getLogger(__name__)
 
 
 class AIModelPlugin(Model):
 
+    _expver = "0000"
+
+    @property
+    def expver(self):
+        if self._expver == "0000":
+            LOG.warning(f"'expver' is not available in this model, using '{self._expver}'.")
+        return self._expver
+
+    @expver.setter
+    def expver(self, value):
+        self._expver = value
+
     def parse_model_args(self, args):
         parser = argparse.ArgumentParser()
 
-        parser.add_argument("--checkpoint")
+        parser.add_argument("--checkpoint", required=not hasattr(self, "download_files"))
         parser.add_argument(
             "--autocast",
             type=str,
-            choices=(
-                "16",
-                "32",
-                "float16",
-                "float32",
-                "bfloat16",
-                "b16",
-            ),
+            choices=sorted(AUTOCAST.keys()),
         )
         args = parser.parse_args(args)
 

src/anemoi/inference/runner.py

@@ -28,7 +28,6 @@
     "bfloat16": torch.bfloat16,
     "float16": torch.float16,
     "float32": torch.float32,
-    None: torch.float16,
 }
 
 
@@ -106,6 +105,10 @@ def run(
         if autocast is None:
             autocast = self.checkpoint.precision
 
+        if autocast is None:
+            LOGGER.warning("No autocast given, using float16")
+            autocast = "16"
+
         autocast = AUTOCAST[autocast]
 
         input_fields = input_fields.sel(**self.checkpoint.select)