set default for implicit multi-threading to a single thread so that d…

…ockerized environment such as binder dont break
cms-opendata-analyses · Feb 27, 2020 · d620d69 · d620d69
1 parent ec826b7
commit d620d69
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/dimuonSpectrum.C b/dimuonSpectrum.C
@@ -20,7 +20,8 @@ float computeInvariantMass(RVec<float>& pt, RVec<float>& eta, RVec<float>& phi,
 
 void dimuonSpectrum() {
     // Enable multi-threading
-    ROOT::EnableImplicitMT();
+    // The default here is set to a single thread. You can choose the number of threads based on your system.
+    ROOT::EnableImplicitMT(1);
 
     // Create dataframe from NanoAOD files
     ROOT::RDataFrame df("Events", "root://eospublic.cern.ch//eos/opendata/cms/derived-data/AOD2NanoAODOutreachTool/Run2012BC_DoubleMuParked_Muons.root");

diff --git a/dimuonSpectrum.ipynb b/dimuonSpectrum.ipynb
@@ -55,7 +55,9 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## Enable multi-threading"
+    "## Enable multi-threading\n",
+    "\n",
+    "The default here is set to a single thread. You can choose the number of threads based on your system."
    ]
   },
   {
@@ -64,7 +66,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "ROOT.ROOT.EnableImplicitMT()"
+    "ROOT.ROOT.EnableImplicitMT(1)"
    ]
   },
   {

diff --git a/dimuonSpectrum.py b/dimuonSpectrum.py
@@ -1,7 +1,8 @@
 import ROOT
 
 # Enable multi-threading
-ROOT.ROOT.EnableImplicitMT()
+# The default here is set to a single thread. You can choose the number of threads based on your system.
+ROOT.ROOT.EnableImplicitMT(1)
 
 # Create dataframe from NanoAOD files
 df = ROOT.RDataFrame("Events", "root://eospublic.cern.ch//eos/opendata/cms/derived-data/AOD2NanoAODOutreachTool/Run2012BC_DoubleMuParked_Muons.root")