update

snowpalm · May 12, 2023 · e133323 · e133323
1 parent 4e35a04
commit e133323
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -40,9 +40,14 @@ def read(fname):
     url="https://github.com/ecmwf-lab/ai-models-panguweather",
     packages=setuptools.find_packages(),
     include_package_data=True,
-    install_requires=['ai-models'],
+    install_requires=["ai-models"],
     zip_safe=True,
     keywords="tool",
+    entry_points={
+        "ai-models.model": [
+            "panguweather = ai_models_panguweather.model:PanguWeather",
+        ]
+    },
     classifiers=[
         "Development Status :: 3 - Alpha",
         "Intended Audience :: Developers",