@@ -363,7 +363,9 @@ <h1>Source code for polaris.job</h1><div class="highlight"><pre>
363363< span class ="sd "> wall_time : str</ span >
364364< span class ="sd "> filesystems : str</ span >
365365< span class ="sd "> """</ span >
366- < span class ="n "> partition_or_queue</ span > < span class ="o "> =</ span > < span class ="n "> config</ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class ="s1 "> 'job'</ span > < span class ="p "> ,</ span > < span class ="n "> partition_or_queue_option</ span > < span class ="p "> )</ span >
366+ < span class ="n "> par_section</ span > < span class ="o "> =</ span > < span class ="n "> config</ span > < span class ="p "> [</ span > < span class ="s1 "> 'parallel'</ span > < span class ="p "> ]</ span >
367+ < span class ="n "> job_section</ span > < span class ="o "> =</ span > < span class ="n "> config</ span > < span class ="p "> [</ span > < span class ="s1 "> 'job'</ span > < span class ="p "> ]</ span >
368+ < span class ="n "> partition_or_queue</ span > < span class ="o "> =</ span > < span class ="n "> job_section</ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class ="n "> partition_or_queue_option</ span > < span class ="p "> )</ span >
367369 < span class ="k "> if</ span > < span class ="n "> partition_or_queue</ span > < span class ="o "> ==</ span > < span class ="s1 "> '<<<default>>>'</ span > < span class ="p "> :</ span >
368370 < span class ="k "> if</ span > < span class ="n "> machine</ span > < span class ="o "> ==</ span > < span class ="s1 "> 'anvil'</ span > < span class ="ow "> and</ span > < span class ="n "> partition_or_queue</ span > < span class ="o "> ==</ span > < span class ="s1 "> 'partition'</ span > < span class ="p "> :</ span >
369371 < span class ="c1 "> # choose the partition based on the number of nodes</ span >
@@ -373,37 +375,35 @@ <h1>Source code for polaris.job</h1><div class="highlight"><pre>
373375 < span class ="n "> partition_or_queue</ span > < span class ="o "> =</ span > < span class ="s1 "> 'acme-medium'</ span >
374376 < span class ="k "> else</ span > < span class ="p "> :</ span >
375377 < span class ="n "> partition_or_queue</ span > < span class ="o "> =</ span > < span class ="s1 "> 'acme-large'</ span >
376- < span class ="k "> elif</ span > < span class ="n "> config </ span > < span class ="o "> .</ span > < span class ="n "> has_option</ span > < span class ="p "> (</ span > < span class =" s1 " > 'parallel' </ span > < span class =" p " > , </ span > < span class ="n "> partitions_or_queues</ span > < span class ="p "> ):</ span >
378+ < span class ="k "> elif</ span > < span class ="n "> par_section </ span > < span class ="o "> .</ span > < span class ="n "> has_option</ span > < span class ="p "> (</ span > < span class ="n "> partitions_or_queues</ span > < span class ="p "> ):</ span >
377379 < span class ="c1 "> # get the first, which is the default</ span >
378- < span class ="n "> partition_or_queue</ span > < span class ="o "> =</ span > < span class ="n "> config</ span > < span class ="o "> .</ span > < span class ="n "> getlist</ span > < span class ="p "> (</ span >
379- < span class ="s1 "> 'parallel'</ span > < span class ="p "> ,</ span > < span class ="n "> partitions_or_queues</ span >
380- < span class ="p "> )[</ span > < span class ="mi "> 0</ span > < span class ="p "> ]</ span >
380+ < span class ="n "> partition_or_queue</ span > < span class ="o "> =</ span > < span class ="n "> par_section</ span > < span class ="o "> .</ span > < span class ="n "> getlist</ span > < span class ="p "> (</ span > < span class ="n "> partitions_or_queues</ span > < span class ="p "> )[</ span > < span class ="mi "> 0</ span > < span class ="p "> ]</ span >
381381 < span class ="k "> else</ span > < span class ="p "> :</ span >
382382 < span class ="n "> partition_or_queue</ span > < span class ="o "> =</ span > < span class ="s1 "> ''</ span >
383383
384- < span class ="n "> qos</ span > < span class ="o "> =</ span > < span class ="n "> config </ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class =" s1 " > 'job' </ span > < span class =" p " > , </ span > < span class ="s1 "> 'qos'</ span > < span class ="p "> )</ span >
384+ < span class ="n "> qos</ span > < span class ="o "> =</ span > < span class ="n "> job_section </ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class ="s1 "> 'qos'</ span > < span class ="p "> )</ span >
385385 < span class ="k "> if</ span > < span class ="n "> qos</ span > < span class ="o "> ==</ span > < span class ="s1 "> '<<<default>>>'</ span > < span class ="p "> :</ span >
386- < span class ="k "> if</ span > < span class ="n "> config </ span > < span class ="o "> .</ span > < span class ="n "> has_option</ span > < span class ="p "> (</ span > < span class =" s1 " > 'parallel' </ span > < span class =" p " > , </ span > < span class ="s1 "> 'qos'</ span > < span class ="p "> ):</ span >
387- < span class ="n "> qos</ span > < span class ="o "> =</ span > < span class ="n "> config </ span > < span class ="o "> .</ span > < span class ="n "> getlist</ span > < span class ="p "> (</ span > < span class =" s1 " > 'parallel' </ span > < span class =" p " > , </ span > < span class ="s1 "> 'qos'</ span > < span class ="p "> )[</ span > < span class ="mi "> 0</ span > < span class ="p "> ]</ span >
386+ < span class ="k "> if</ span > < span class ="n "> par_section </ span > < span class ="o "> .</ span > < span class ="n "> has_option</ span > < span class ="p "> (</ span > < span class ="s1 "> 'qos'</ span > < span class ="p "> ):</ span >
387+ < span class ="n "> qos</ span > < span class ="o "> =</ span > < span class ="n "> par_section </ span > < span class ="o "> .</ span > < span class ="n "> getlist</ span > < span class ="p "> (</ span > < span class ="s1 "> 'qos'</ span > < span class ="p "> )[</ span > < span class ="mi "> 0</ span > < span class ="p "> ]</ span >
388388 < span class ="k "> else</ span > < span class ="p "> :</ span >
389389 < span class ="n "> qos</ span > < span class ="o "> =</ span > < span class ="s1 "> ''</ span >
390390
391- < span class ="n "> constraint</ span > < span class ="o "> =</ span > < span class ="n "> config </ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class =" s1 " > 'job' </ span > < span class =" p " > , </ span > < span class ="s1 "> 'constraint'</ span > < span class ="p "> )</ span >
391+ < span class ="n "> constraint</ span > < span class ="o "> =</ span > < span class ="n "> job_section </ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class ="s1 "> 'constraint'</ span > < span class ="p "> )</ span >
392392 < span class ="k "> if</ span > < span class ="n "> constraint</ span > < span class ="o "> ==</ span > < span class ="s1 "> '<<<default>>>'</ span > < span class ="p "> :</ span >
393- < span class ="k "> if</ span > < span class ="n "> config </ span > < span class ="o "> .</ span > < span class ="n "> has_option</ span > < span class ="p "> (</ span > < span class =" s1 " > 'parallel' </ span > < span class =" p " > , </ span > < span class ="s1 "> 'constraints'</ span > < span class ="p "> ):</ span >
394- < span class ="n "> constraint</ span > < span class ="o "> =</ span > < span class ="n "> config </ span > < span class ="o "> .</ span > < span class ="n "> getlist</ span > < span class ="p "> (</ span > < span class =" s1 " > 'parallel' </ span > < span class =" p " > , </ span > < span class ="s1 "> 'constraints'</ span > < span class ="p "> )[</ span > < span class ="mi "> 0</ span > < span class ="p "> ]</ span >
393+ < span class ="k "> if</ span > < span class ="n "> par_section </ span > < span class ="o "> .</ span > < span class ="n "> has_option</ span > < span class ="p "> (</ span > < span class ="s1 "> 'constraints'</ span > < span class ="p "> ):</ span >
394+ < span class ="n "> constraint</ span > < span class ="o "> =</ span > < span class ="n "> par_section </ span > < span class ="o "> .</ span > < span class ="n "> getlist</ span > < span class ="p "> (</ span > < span class ="s1 "> 'constraints'</ span > < span class ="p "> )[</ span > < span class ="mi "> 0</ span > < span class ="p "> ]</ span >
395395 < span class ="k "> else</ span > < span class ="p "> :</ span >
396396 < span class ="n "> constraint</ span > < span class ="o "> =</ span > < span class ="s1 "> ''</ span >
397397
398- < span class ="k "> if</ span > < span class ="n "> config </ span > < span class ="o "> .</ span > < span class ="n "> has_option</ span > < span class ="p "> (</ span > < span class =" s1 " > 'parallel' </ span > < span class =" p " > , </ span > < span class ="s1 "> 'gpus_per_node'</ span > < span class ="p "> ):</ span >
399- < span class ="n "> gpus_per_node</ span > < span class ="o "> =</ span > < span class ="n "> config </ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class =" s1 " > 'parallel' </ span > < span class =" p " > , </ span > < span class ="s1 "> 'gpus_per_node'</ span > < span class ="p "> )</ span >
398+ < span class ="k "> if</ span > < span class ="n "> par_section </ span > < span class ="o "> .</ span > < span class ="n "> has_option</ span > < span class ="p "> (</ span > < span class ="s1 "> 'gpus_per_node'</ span > < span class ="p "> ):</ span >
399+ < span class ="n "> gpus_per_node</ span > < span class ="o "> =</ span > < span class ="n "> par_section </ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class ="s1 "> 'gpus_per_node'</ span > < span class ="p "> )</ span >
400400 < span class ="k "> else</ span > < span class ="p "> :</ span >
401401 < span class ="n "> gpus_per_node</ span > < span class ="o "> =</ span > < span class ="s1 "> ''</ span >
402402
403- < span class ="n "> wall_time</ span > < span class ="o "> =</ span > < span class ="n "> config </ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class =" s1 " > 'job' </ span > < span class =" p " > , </ span > < span class ="s1 "> 'wall_time'</ span > < span class ="p "> )</ span >
403+ < span class ="n "> wall_time</ span > < span class ="o "> =</ span > < span class ="n "> job_section </ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class ="s1 "> 'wall_time'</ span > < span class ="p "> )</ span >
404404
405- < span class ="k "> if</ span > < span class ="n "> config </ span > < span class ="o "> .</ span > < span class ="n "> has_option</ span > < span class ="p "> (</ span > < span class =" s1 " > 'job' </ span > < span class =" p " > , </ span > < span class ="s1 "> 'filesystems'</ span > < span class ="p "> ):</ span >
406- < span class ="n "> filesystems</ span > < span class ="o "> =</ span > < span class ="n "> config </ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class =" s1 " > 'job' </ span > < span class =" p " > , </ span > < span class ="s1 "> 'filesystems'</ span > < span class ="p "> )</ span >
405+ < span class ="k "> if</ span > < span class ="n "> job_section </ span > < span class ="o "> .</ span > < span class ="n "> has_option</ span > < span class ="p "> (</ span > < span class ="s1 "> 'filesystems'</ span > < span class ="p "> ):</ span >
406+ < span class ="n "> filesystems</ span > < span class ="o "> =</ span > < span class ="n "> job_section </ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class ="s1 "> 'filesystems'</ span > < span class ="p "> )</ span >
407407 < span class ="k "> else</ span > < span class ="p "> :</ span >
408408 < span class ="n "> filesystems</ span > < span class ="o "> =</ span > < span class ="s1 "> ''</ span >
409409
0 commit comments