From eac8eeaf84fba3d76bd4ef852ef663fded94f811 Mon Sep 17 00:00:00 2001
From: tourniert <theo.tournier@meteo.fr>
Date: Fri, 6 Sep 2024 14:51:21 +0200
Subject: [PATCH] set strict dependency | add add joblib + torch_geometric

---
 requirements.txt | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 39abc173..c8d4486f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,7 +3,7 @@ wandb>=0.13.10
 matplotlib>=3.7.0
 scipy>=1.10.0
 pytorch-lightning>=2.1.2
-lightning
+lightning==2.2.2
 shapely>=2.0.1
 networkx>=3.0
 Cartopy>=0.22.0
@@ -12,14 +12,14 @@ tueplots>=0.0.8
 plotly>=5.15.0
 tornado>=6.3.3
 cython>=3
-cfgrib
+cfgrib==0.9.14.0
 dataclasses-json==0.6.4
-xarray
+xarray==2024.7.0
 argparse-dataclass==2.0.0
 tensorboard==2.13.0
 typer==0.9.0
 netCDF4==1.6.5
-tensorboard-plugin-profile
+tensorboard-plugin-profile==2.17.0
 torch-tb-profiler==0.4.1
 einops==0.7.0
 torchinfo==1.8.0
@@ -29,7 +29,7 @@ coverage==7.6.1
 onnx==1.16.1
 onnxruntime==1.18.1
 onnxruntime-gpu==1.18.1
-onnxscript
+onnxscript==0.1.0.dev20240905
 monai==1.3.1
 gif==23.3.0
 scikit-image==0.24.0
@@ -37,3 +37,5 @@ typer==0.9.0
 transformers==4.44.2
 tensorflow==2.13.1
 py-spy
+torch_geometric==2.3.1
+joblib==1.4.2
\ No newline at end of file