diff --git a/TurTLE/_code.py b/TurTLE/_code.py index 89ac3e3667363cd3392a8106f66bf26cd066ee24..64df8f074b4a7cc723a5d0f51f62bfd43c4bb61d 100644 --- a/TurTLE/_code.py +++ b/TurTLE/_code.py @@ -664,8 +664,14 @@ class _code(_base): # output file if not type(out_file) == type(None): script_file.write('#SBATCH -o ' + out_file + '\n') + + # set up environment + script_file.write('#SBATCH --get-user-env\n') script_file.write('#SBATCH --partition={0}\n'.format( self.host_info['environment'])) + if 'account' in self.host_info.keys(): + script_file.write('#SBATCH --account={0}\n'.format( + self.host_info['account'])) nb_cpus_per_node = self.host_info['deltanprocs'] assert isinstance(nb_cpus_per_node, int) \ diff --git a/pc_host_info.py b/pc_host_info.py index 1c5eee58ed529fb496a2863775391a7b14058cf7..4c6b3816961947c111162a65810e343c8559d44e 100644 --- a/pc_host_info.py +++ b/pc_host_info.py @@ -29,8 +29,9 @@ host_info = {'type' : 'pc'} # {'type' : info_template_type, # 'MPI' : info_template_MPI, # 'environment' : info_template_environment, -# 'deltanprocs' : info_template_deltanprocs -# 'mail_address': info_template_mail_address} +# 'deltanprocs' : info_template_deltanprocs, +# 'mail_address': info_template_mail_address, +# 'account' : info_template_account} # info_template_type can be one of: # 'pc' --- jobs run interactively @@ -50,3 +51,6 @@ host_info = {'type' : 'pc'} # info_template_mail_address, relevant for clusters, # is the contact e-mail address placed in the job scripts. + +# info_template_account, relevant for some clusters, +# is the name of the account to be budgeted for the job.