update

jmduarte · May 14, 2024 · 129e553 · 129e553
1 parent 0692e13
commit 129e553
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -11,11 +11,11 @@ USER jovyan
 
 RUN mamba install -c conda-forge uproot xrootd root
 
-RUN pip install --no-cache-dir 'xgboost==2.0.3' 'scikit-learn==1.4.1' 'spektral==1.3.1' 'gdown==5.1.0' 'mplhep==0.3.43' && \
+RUN pip install --no-cache-dir 'xgboost==2.0.3' 'spektral==1.3.1' 'gdown==5.1.0' 'mplhep==0.3.43' && \
     fix-permissions /opt/conda && \
     fix-permissions /home/jovyan
 
-RUN pip install --no-cache-dir --no-index pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.1.0+cu118.html && \
+RUN pip install --no-cache-dir --no-index pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.1.2+cu118.html && \
     pip install --no-cache-dir typing-extensions --upgrade
 
 # RUN pip install --no-cache-dir 'jetnet==0.2.2'