update dependencies

Author: b8raoult

ai_models_graphcast/__init__.py

@@ -5,4 +5,4 @@
 # granted to it by virtue of its status as an intergovernmental organisation
 # nor does it submit to any jurisdiction.
 
-__version__ = "0.0.6"
+__version__ = "0.0.7"

setup.py

@@ -43,7 +43,7 @@ def read(fname):
     include_package_data=True,
     # JAX requirements are in requirements.txt
     install_requires=[
-        "ai-models>=0.3.6",
+        "ai-models>=0.4.0",
         "dm-tree",
         "dm-haiku==0.0.10",
     ],