updates

Author: jmduarte

Dockerfile

@@ -12,6 +12,6 @@ USER jovyan
 RUN mamba install -c conda-forge uproot xrootd
 
 RUN pip install --no-cache-dir 'xgboost==3.0.5' 'gdown==5.2.0' 'mplhep==0.4.1' 'jetnet==0.2.5' 'tables==3.10.2' 'torch_geometric' && \
-    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.6.0+cu126.html && \
+    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.2.1+cu121.html && \
     fix-permissions /opt/conda && \
     fix-permissions /home/jovyan

notebooks/06_Graph_Data_GNN.ipynb

@@ -12,21 +12,15 @@
     "\n",
     "Depending on the environment used, you may need to install additional libaries:\n",
     "```ipython\n",
-    "!pip install uproot torch_geometric mplhep\n",
-    "!pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.6.0+cu126.html \n",
+    "!pip install xrootd torch_geometric mplhep\n",
+    "!pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.2.1+cu121.html \n",
     "```\n",
     "\n",
     "Data can also be downloaded locally:\n",
     "```ipython\n",
     "!wget https://opendata.cern.ch/record/12102/files/assets/cms/datascience/HiggsToBBNtupleProducerTool/HiggsToBBNTuple_HiggsToBB_QCD_RunII_13TeV_MC/train/ntuple_merged_10.root\n",
     "!wget https://opendata.cern.ch/record/12102/files/assets/cms/datascience/HiggsToBBNtupleProducerTool/HiggsToBBNTuple_HiggsToBB_QCD_RunII_13TeV_MC/test/ntuple_merged_0.root\n",
-    "```\n",
-    "\n",
-    "on DataHub two files have been copied locally to \n",
-    "```\n",
-    "/higgs/train/ntuple_merged_10.root\n",
-    "/higgs/test/ntuple_merged_0.root\n",
-    "````"
+    "```"
    ]
   },
   {
@@ -93,8 +87,8 @@
     "local = False\n",
     "\n",
     "if local:\n",
-    "    file_names = [\"/higgs/train/ntuple_merged_10.root\"]\n",
-    "    file_names_test = [\"/higgs/test/ntuple_merged_0.root\"]\n",
+    "    file_names = [\"ntuple_merged_10.root\"]\n",
+    "    file_names_test = [\"ntuple_merged_0.root\"]\n",
     "else:\n",
     "    file_names = [\n",
     "        \"root://eospublic.cern.ch//eos/opendata/cms/datascience/HiggsToBBNtupleProducerTool/HiggsToBBNTuple_HiggsToBB_QCD_RunII_13TeV_MC/train/ntuple_merged_10.root\"\n",