From 9b28685d654a78e55d3da99bef62beacaea630d6 Mon Sep 17 00:00:00 2001
From: Baudouin Raoult <baudouin.raoult@ecmwf.int>
Date: Tue, 13 Feb 2024 07:44:00 +0000
Subject: [PATCH] update dependencies

---
 ai_models_graphcast/__init__.py | 2 +-
 setup.py                        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/ai_models_graphcast/__init__.py b/ai_models_graphcast/__init__.py
index 1dc7578..dddd955 100644
--- a/ai_models_graphcast/__init__.py
+++ b/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"
diff --git a/setup.py b/setup.py
index 0e45934..840e3f7 100644
--- a/setup.py
+++ b/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",
     ],