APR4c_ERF_submit_to_cluster.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
import os.path as op
import numpy as np
import pickle
from warnings import filterwarnings
from sys import argv
from stormdb.cluster import ClusterBatch
from stormdb.access import Query

project = 'MINDLAB2020_MEG-AuditoryPatternRecognition'
os.environ['MINDLABPROJ']=project
os.environ['MNE_ROOT']='~/miniconda3/envs/mne' # for surfer
os.environ['MESA_GL_VERSION_OVERRIDE'] = '3.2'
script_dir = '/projects/{}/scripts/working_memory/'.format(project)

qr = Query(project)
sub_codes = qr.get_subjects()

subNs = range(11,91)
#subNs = [12,23,40,58,76]
cb = ClusterBatch(project)
for s in subNs:
    #sub = sub_codes[s-1]
    submit_cmd = 'python {}APR4a_ERF_analyses.py {}'.format(script_dir,s)
    cb.add_job(cmd=submit_cmd, queue='short.q',n_threads = 1, cleanup = False)

cb.submit()