From 5e81d494a5262f9b8b61331c1777a5f68f16c5f4 Mon Sep 17 00:00:00 2001
From: Thomas Kluyver <thomas.kluyver@xfel.eu>
Date: Fri, 9 Apr 2021 17:33:32 +0200
Subject: [PATCH] Make max_nodes a parameter for AGIPD correction notebook
---
notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
index 8ce400170..f49c2c540 100644
--- a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
+++ b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
@@ -87,10 +87,11 @@
"n_cores_correct = 16 # Number of chunks to be processed in parallel\n",
"n_cores_files = 4 # Number of files to be processed in parallel\n",
"sequences_per_node = 2 # number of sequence files per cluster node if run as slurm job, set to 0 to not run SLURM parallel\n",
+ "max_nodes = 8 # Maximum number of Slurm jobs to split correction work into\n",
"\n",
- "def balance_sequences(in_folder, run, sequences, sequences_per_node, karabo_da):\n",
+ "def balance_sequences(in_folder, run, sequences, sequences_per_node, karabo_da, max_nodes):\n",
" from xfel_calibrate.calibrate import balance_sequences as bs\n",
- " return bs(in_folder, run, sequences, sequences_per_node, karabo_da)"
+ " return bs(in_folder, run, sequences, sequences_per_node, karabo_da, max_nodes=max_nodes)"
]
},
{
--
GitLab