diff --git a/pyproject.toml b/pyproject.toml
index 57fc0ff..e2a0ac1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -50,7 +50,10 @@ dynamic = [
   "version",
 ]
 dependencies = [
-  "anemoi-datasets",
+  "anemoi-datasets[data]>=0.3.3",
+  "torch>=2.2",
+  "torch-geometric>=2.3.1,<2.5",
+  "anemoi-utils>=0.1.3",
 ]
 
 optional-dependencies.all = [