import sys 
import os
import subprocess

sim_pack = sys.argv[1]
task = sys.argv[2]
com_cap = sys.argv[3]
prec = sys.argv[4]

N_arr = range(6, 36, 2)

for N in N_arr:
    with open('{}_{}_{}_{}.sh'.format(task, com_cap, prec, N), 'a') as job_file:
        job_file.write('#!/bin/bash -l\n')
        job_file.write('. /app/etc/profile.d/conda.sh\n')

        if com_cap == 'st' or com_cap == 'mt':
            job_file.write('conda activate sim_{}\n'.format(sim_pack))

        elif com_cap == 'gpu':
            job_file.write('conda activate sim_{}_gpu\n'.format(sim_pack))

        job_file.write('python3 /home/{}_n{}.py\n'.format(task, N))