updated slurm array script for jeanZay cluster

scripts/launchSlurmArray.py

@@ -12,7 +12,7 @@ def launchSlurmArray(names, commands, jobName, device, nbHours, limit) :
 
   commandList = " ".join(commands)
 
-  if hostname == "jean-zay1" :
+  if "jean-zay" in hostname :
     print("""#! /usr/bin/env bash
 
 #SBATCH --array=0-{}%{}
@@ -24,8 +24,8 @@ def launchSlurmArray(names, commands, jobName, device, nbHours, limit) :
 #SBATCH --cpus-per-task=10
 #SBATCH --gres=gpu:1
 #SBATCH --hint=nomultithread
-#SBATCH --partition=gpu_p1
 #SBATCH --qos={}
+#SBATCH --constraint=v100-16g
 #SBATCH --time={}:00:00
 
 module purge
@@ -56,7 +56,7 @@ cat $tmpFile >> $newErr
 rm $tmpFile
 
 eval "${{commands[$SLURM_ARRAY_TASK_ID]}}"
-""".format(len(names)-1, limit, jobName, "qos_gpu-t4" if nbHours > 20 else "qos_gpu-t3", nbHours, " ".join(names), " ".join(commands)), file=sFile)
+""".format(len(names)-1, limit, jobName, "qos_gpu-t4" if int(nbHours) > 20 else "qos_gpu-t3", nbHours, " ".join(names), " ".join(commands)), file=sFile)
     sFile.close()
   elif hostname == "sms.liscluster" :
     print('''#! /usr/bin/env bash