set strict dependency | add add joblib + torch_geometric

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,11 +29,13 @@ 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
 typer==0.9.0
 transformers==4.44.2
 tensorflow==2.13.1
 py-spy
+torch_geometric==2.3.1
+joblib==1.4.2