@@ -224,10 +224,8 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
224
224
< span class ="kn "> from</ span > < span class ="nn "> .</ span > < span class ="kn "> import</ span > < span class ="n "> __version__</ span >
225
225
< span class ="kn "> from</ span > < span class ="nn "> .format</ span > < span class ="kn "> import</ span > < span class ="n "> format_with_config</ span >
226
226
< span class ="kn "> from</ span > < span class ="nn "> .templates</ span > < span class ="kn "> import</ span > < span class ="p "> (</ span >
227
- < span class ="n "> EXTRACT_RESULTS</ span > < span class ="p "> ,</ span >
228
227
< span class ="n "> JOB_INTERACTIVE_TEMPLATE</ span > < span class ="p "> ,</ span >
229
228
< span class ="n "> JOB_SCRIPT_TEMPLATE</ span > < span class ="p "> ,</ span >
230
- < span class ="n "> SAVE_RESULTS</ span > < span class ="p "> ,</ span >
231
229
< span class ="n "> VARS_TEMPLATE</ span > < span class ="p "> ,</ span >
232
230
< span class ="p "> )</ span >
233
231
@@ -237,7 +235,6 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
237
235
< span class ="o "> *</ span > < span class ="p "> ,</ span >
238
236
< span class ="n "> src</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="o "> =</ span > < span class ="s2 "> ""</ span > < span class ="p "> ,</ span >
239
237
< span class ="n "> assets</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="o "> =</ span > < span class ="s2 "> ""</ span > < span class ="p "> ,</ span >
240
- < span class ="n "> dataset</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="o "> =</ span > < span class ="s2 "> ""</ span > < span class ="p "> ,</ span >
241
238
< span class ="n "> on_run</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="o "> =</ span > < span class ="s2 "> ""</ span > < span class ="p "> ,</ span >
242
239
< span class ="n "> on_run_resume</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="o "> =</ span > < span class ="s2 "> ""</ span > < span class ="p "> ,</ span >
243
240
< span class ="n "> setup</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="o "> =</ span > < span class ="s2 "> ""</ span > < span class ="p "> ,</ span >
@@ -248,7 +245,6 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
248
245
< span class ="n "> submit</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> True</ span > < span class ="p "> ,</ span >
249
246
< span class ="n "> interactive</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> ,</ span >
250
247
< span class ="n "> resubmit_limit</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="o "> =</ span > < span class ="mi "> 64</ span > < span class ="p "> ,</ span >
251
- < span class ="n "> results_sync_method</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="o "> =</ span > < span class ="s2 "> "symlink"</ span > < span class ="p "> ,</ span >
252
248
< span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="nb "> str</ span > < span class ="p "> :</ span >
253
249
< span class ="w "> </ span > < span class ="sd "> """Submits job.</ span >
254
250
@@ -257,9 +253,7 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
257
253
< span class ="sd "> Args:</ span >
258
254
< span class ="sd "> job_dir (str):</ span >
259
255
< span class ="sd "> Path to directory to keep all job files including</ span >
260
- < span class ="sd "> ``src.tar``, ``assets.tar``, auto-generated ``job.sh``, and</ span >
261
- < span class ="sd "> results. Note that the ``dataset`` will not be copied and</ span >
262
- < span class ="sd "> will remain in its original path.</ span >
256
+ < span class ="sd "> ``src.tar``, ``assets.tar``, and auto-generated ``job.sh``.</ span >
263
257
< span class ="sd "> src (str):</ span >
264
258
< span class ="sd "> Path to directory containing only source code.</ span >
265
259
< span class ="sd "> These will be archived in ``$JOB_DIR/src.tar`` and</ span >
@@ -268,10 +262,6 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
268
262
< span class ="sd "> Path to directory containing additional assets.</ span >
269
263
< span class ="sd "> These will be archived in ``$JOB_DIR/assets.tar`` and</ span >
270
264
< span class ="sd "> extracted during job run into ``$SLURM_TMPDIR/assets``.</ span >
271
- < span class ="sd "> dataset (str):</ span >
272
- < span class ="sd "> Path to ``.tar`` archive of dataset. This will be copied and</ span >
273
- < span class ="sd "> extracted on the local filesystem of the compute node,</ span >
274
- < span class ="sd "> ``$SLURM_TMPDIR``.</ span >
275
265
< span class ="sd "> on_run (str):</ span >
276
266
< span class ="sd "> Bash code executed in "on_run" stage, but only for new jobs</ span >
277
267
< span class ="sd "> that are running for the first time.</ span >
@@ -297,7 +287,7 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
297
287
< span class ="sd "> Dictionary of options to pass to sbatch.</ span >
298
288
< span class ="sd "> cleanup_seconds (int):</ span >
299
289
< span class ="sd "> Interrupts a job n seconds before timeout to run cleanup</ span >
300
- < span class ="sd "> tasks (teardown, save_results, auto-schedule new job).</ span >
290
+ < span class ="sd "> tasks (teardown, auto-schedule new job).</ span >
301
291
< span class ="sd "> Default is 120 seconds.</ span >
302
292
< span class ="sd "> submit (bool):</ span >
303
293
< span class ="sd "> Submit created job to scheduler. Set this to ``False`` if</ span >
@@ -309,14 +299,6 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
309
299
< span class ="sd "> Maximum number of times to auto-submit a job for "resume".</ span >
310
300
< span class ="sd "> (Not entirely unlike submitting a resume for a job.)</ span >
311
301
< span class ="sd "> Default is 64 resubmissions.</ span >
312
- < span class ="sd "> results_sync_method (str):</ span >
313
- < span class ="sd "> Choices: "rsync", "symlink", or "targz".</ span >
314
-
315
- < span class ="sd "> - rsync: Sync results directory via rsync.</ span >
316
- < span class ="sd "> - symlink: Directly symlink results directory.</ span >
317
- < span class ="sd "> - targz: Extract/archive results directory into .tar.gz.</ span >
318
-
319
- < span class ="sd "> Default is ``"symlink"``.</ span >
320
302
321
303
< span class ="sd "> Returns:</ span >
322
304
< span class ="sd "> Path to the newly created job directory.</ span >
@@ -335,10 +317,8 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
335
317
< span class ="n "> setup_resume</ span > < span class ="o "> =</ span > < span class ="n "> setup_resume</ span > < span class ="p "> ,</ span >
336
318
< span class ="n "> teardown</ span > < span class ="o "> =</ span > < span class ="n "> teardown</ span > < span class ="p "> ,</ span >
337
319
< span class ="n "> job_dir</ span > < span class ="o "> =</ span > < span class ="n "> job_dir</ span > < span class ="p "> ,</ span >
338
- < span class ="n "> dataset</ span > < span class ="o "> =</ span > < span class ="n "> dataset</ span > < span class ="p "> ,</ span >
339
320
< span class ="n "> cleanup_seconds</ span > < span class ="o "> =</ span > < span class ="n "> cleanup_seconds</ span > < span class ="p "> ,</ span >
340
321
< span class ="n "> resubmit_limit</ span > < span class ="o "> =</ span > < span class ="n "> resubmit_limit</ span > < span class ="p "> ,</ span >
341
- < span class ="n "> results_sync_method</ span > < span class ="o "> =</ span > < span class ="n "> results_sync_method</ span > < span class ="p "> ,</ span >
342
322
< span class ="p "> )</ span >
343
323
< span class ="n "> _write_script</ span > < span class ="p "> (</ span > < span class ="n "> job_path</ span > < span class ="p "> ,</ span > < span class ="n "> job_script_str</ span > < span class ="p "> )</ span >
344
324
@@ -365,30 +345,21 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
365
345
< span class ="n "> setup_resume</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span >
366
346
< span class ="n "> teardown</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span >
367
347
< span class ="n "> job_dir</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span >
368
- < span class ="n "> dataset</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span >
369
348
< span class ="n "> cleanup_seconds</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span >
370
349
< span class ="n "> resubmit_limit</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span >
371
- < span class ="n "> results_sync_method</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span >
372
350
< span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="nb "> str</ span > < span class ="p "> :</ span >
373
351
< span class ="w "> </ span > < span class ="sd "> """Returns source for job script."""</ span >
374
352
< span class ="n "> job_dir</ span > < span class ="o "> =</ span > < span class ="n "> _expand_path</ span > < span class ="p "> (</ span > < span class ="n "> job_dir</ span > < span class ="p "> )</ span >
375
- < span class ="n "> dataset</ span > < span class ="o "> =</ span > < span class ="n "> _expand_path</ span > < span class ="p "> (</ span > < span class ="n "> dataset</ span > < span class ="p "> )</ span >
376
353
377
354
< span class ="n "> vars_str</ span > < span class ="o "> =</ span > < span class ="n "> VARS_TEMPLATE</ span > < span class ="o "> .</ span > < span class ="n "> format</ span > < span class ="p "> (</ span >
378
355
< span class ="n "> easy_slurm_version</ span > < span class ="o "> =</ span > < span class ="n "> __version__</ span > < span class ="p "> ,</ span >
379
356
< span class ="n "> job_dir</ span > < span class ="o "> =</ span > < span class ="n "> job_dir</ span > < span class ="p "> ,</ span >
380
- < span class ="n "> dataset_path</ span > < span class ="o "> =</ span > < span class ="n "> dataset</ span > < span class ="p "> ,</ span >
381
357
< span class ="n "> resubmit_limit</ span > < span class ="o "> =</ span > < span class ="n "> resubmit_limit</ span > < span class ="p "> ,</ span >
382
358
< span class ="p "> )</ span >
383
359
384
- < span class ="n "> extract_results</ span > < span class ="o "> =</ span > < span class ="n "> EXTRACT_RESULTS</ span > < span class ="p "> [</ span > < span class ="n "> results_sync_method</ span > < span class ="p "> ]</ span >
385
- < span class ="n "> save_results</ span > < span class ="o "> =</ span > < span class ="n "> SAVE_RESULTS</ span > < span class ="p "> [</ span > < span class ="n "> results_sync_method</ span > < span class ="p "> ]</ span >
386
-
387
360
< span class ="n "> setup</ span > < span class ="o "> =</ span > < span class ="n "> _fix_indent</ span > < span class ="p "> (</ span > < span class ="n "> setup</ span > < span class ="p "> ,</ span > < span class ="mi "> 1</ span > < span class ="p "> )</ span >
388
361
< span class ="n "> setup_resume</ span > < span class ="o "> =</ span > < span class ="n "> _fix_indent</ span > < span class ="p "> (</ span > < span class ="n "> setup_resume</ span > < span class ="p "> ,</ span > < span class ="mi "> 1</ span > < span class ="p "> )</ span >
389
362
< span class ="n "> teardown</ span > < span class ="o "> =</ span > < span class ="n "> _fix_indent</ span > < span class ="p "> (</ span > < span class ="n "> teardown</ span > < span class ="p "> ,</ span > < span class ="mi "> 1</ span > < span class ="p "> )</ span >
390
- < span class ="n "> extract_results</ span > < span class ="o "> =</ span > < span class ="n "> _fix_indent</ span > < span class ="p "> (</ span > < span class ="n "> extract_results</ span > < span class ="p "> ,</ span > < span class ="mi "> 1</ span > < span class ="p "> )</ span >
391
- < span class ="n "> save_results</ span > < span class ="o "> =</ span > < span class ="n "> _fix_indent</ span > < span class ="p "> (</ span > < span class ="n "> save_results</ span > < span class ="p "> ,</ span > < span class ="mi "> 1</ span > < span class ="p "> )</ span >
392
363
393
364
< span class ="n "> fix_quotes</ span > < span class ="o "> =</ span > < span class ="k "> lambda</ span > < span class ="n "> x</ span > < span class ="p "> :</ span > < span class ="n "> _quote_single_quotes</ span > < span class ="p "> (</ span > < span class ="n "> x</ span > < span class ="o "> .</ span > < span class ="n "> strip</ span > < span class ="p "> ())</ span >
394
365
< span class ="n "> on_run</ span > < span class ="o "> =</ span > < span class ="n "> fix_quotes</ span > < span class ="p "> (</ span > < span class ="n "> on_run</ span > < span class ="p "> )</ span >
@@ -406,8 +377,6 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
406
377
< span class ="n "> setup</ span > < span class ="o "> =</ span > < span class ="n "> setup</ span > < span class ="p "> ,</ span >
407
378
< span class ="n "> setup_resume</ span > < span class ="o "> =</ span > < span class ="n "> setup_resume</ span > < span class ="p "> ,</ span >
408
379
< span class ="n "> teardown</ span > < span class ="o "> =</ span > < span class ="n "> teardown</ span > < span class ="p "> ,</ span >
409
- < span class ="n "> extract_results</ span > < span class ="o "> =</ span > < span class ="n "> extract_results</ span > < span class ="p "> ,</ span >
410
- < span class ="n "> save_results</ span > < span class ="o "> =</ span > < span class ="n "> save_results</ span > < span class ="p "> ,</ span >
411
380
< span class ="p "> )</ span > </ div >
412
381
413
382
0 commit comments