Change outputs to plugin format

ai_models/__main__.py

@@ -97,7 +97,7 @@ def _main():
         "--output",
         default="file",
         help="Where to output the results",
-        choices=available_outputs(),
+        choices=sorted(available_outputs()),
     )
 
     parser.add_argument(

ai_models/outputs/__init__.py

@@ -9,6 +9,7 @@
 import logging
 
 import climetlab as cml
+import entrypoints
 import numpy as np
 
 LOG = logging.getLogger(__name__)
@@ -112,18 +113,15 @@ def write(self, *args, **kwargs):
         pass
 
 
-OUTPUTS = dict(
-    file=FileOutput,
-    none=NoneOutput,
-)
-
-
 def get_output(name, owner, *args, **kwargs):
-    result = OUTPUTS[name](owner, *args, **kwargs)
+    result = available_outputs()[name].load()(owner, *args, **kwargs)
     if kwargs.get("hindcast_reference_year") is not None:
         result = HindcastReLabel(owner, result, **kwargs)
     return result
 
 
 def available_outputs():
-    return sorted(OUTPUTS.keys())
+    result = {}
+    for e in entrypoints.get_group_all("ai_models.output"):
+        result[e.name] = e
+    return result

setup.py

@@ -64,6 +64,10 @@ def read(fname):
             "mars=ai_models.inputs:MarsInput",
             "cds=ai_models.inputs:CdsInput",
         ],
+        "ai_models.output": [
+            "file=ai_models.outputs:FileOutput",
+            "none=ai_models.outputs:NoneOutput",
+        ],
     },
     classifiers=[
         "Development Status :: 3 - Alpha",