use pyproject

ecmwf-lab · May 6, 2024 · 192db96 · 192db96
1 parent f044b59
commit 192db96
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -62,3 +62,13 @@ ai-models = "ai_models.__main__:main"
 
 [tool.setuptools_scm]
 version_file = "src/ai_models/_version.py"
+
+[project.entry-points."ai_models.input"]
+file = "ai_models.inputs:FileInput"
+mars = "ai_models.inputs:MarsInput"
+cds = "ai_models.inputs:CdsInput"
+opendata = "ai_models.inputs:OpenDataInput"
+
+[project.entry-points."ai_models.output"]
+file = "ai_models.outputs:FileOutput"
+none = "ai_models.outputs:NoneOutput"