Skip to content

Commit cf11966

Browse files
committed
deploy: b03f0d5
1 parent 5dba3e4 commit cf11966

File tree

5 files changed

+9
-57
lines changed

5 files changed

+9
-57
lines changed

_modules/easy_slurm/jobs.html

Lines changed: 2 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -224,10 +224,8 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
224224
<span class="kn">from</span> <span class="nn">.</span> <span class="kn">import</span> <span class="n">__version__</span>
225225
<span class="kn">from</span> <span class="nn">.format</span> <span class="kn">import</span> <span class="n">format_with_config</span>
226226
<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>
228227
<span class="n">JOB_INTERACTIVE_TEMPLATE</span><span class="p">,</span>
229228
<span class="n">JOB_SCRIPT_TEMPLATE</span><span class="p">,</span>
230-
<span class="n">SAVE_RESULTS</span><span class="p">,</span>
231229
<span class="n">VARS_TEMPLATE</span><span class="p">,</span>
232230
<span class="p">)</span>
233231

@@ -237,7 +235,6 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
237235
<span class="o">*</span><span class="p">,</span>
238236
<span class="n">src</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;&quot;</span><span class="p">,</span>
239237
<span class="n">assets</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;&quot;</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">&quot;&quot;</span><span class="p">,</span>
241238
<span class="n">on_run</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;&quot;</span><span class="p">,</span>
242239
<span class="n">on_run_resume</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;&quot;</span><span class="p">,</span>
243240
<span class="n">setup</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;&quot;</span><span class="p">,</span>
@@ -248,7 +245,6 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
248245
<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>
249246
<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>
250247
<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">&quot;symlink&quot;</span><span class="p">,</span>
252248
<span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">str</span><span class="p">:</span>
253249
<span class="w"> </span><span class="sd">&quot;&quot;&quot;Submits job.</span>
254250

@@ -257,9 +253,7 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
257253
<span class="sd"> Args:</span>
258254
<span class="sd"> job_dir (str):</span>
259255
<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>
263257
<span class="sd"> src (str):</span>
264258
<span class="sd"> Path to directory containing only source code.</span>
265259
<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>
268262
<span class="sd"> Path to directory containing additional assets.</span>
269263
<span class="sd"> These will be archived in ``$JOB_DIR/assets.tar`` and</span>
270264
<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>
275265
<span class="sd"> on_run (str):</span>
276266
<span class="sd"> Bash code executed in &quot;on_run&quot; stage, but only for new jobs</span>
277267
<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>
297287
<span class="sd"> Dictionary of options to pass to sbatch.</span>
298288
<span class="sd"> cleanup_seconds (int):</span>
299289
<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>
301291
<span class="sd"> Default is 120 seconds.</span>
302292
<span class="sd"> submit (bool):</span>
303293
<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>
309299
<span class="sd"> Maximum number of times to auto-submit a job for &quot;resume&quot;.</span>
310300
<span class="sd"> (Not entirely unlike submitting a resume for a job.)</span>
311301
<span class="sd"> Default is 64 resubmissions.</span>
312-
<span class="sd"> results_sync_method (str):</span>
313-
<span class="sd"> Choices: &quot;rsync&quot;, &quot;symlink&quot;, or &quot;targz&quot;.</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 ``&quot;symlink&quot;``.</span>
320302

321303
<span class="sd"> Returns:</span>
322304
<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>
335317
<span class="n">setup_resume</span><span class="o">=</span><span class="n">setup_resume</span><span class="p">,</span>
336318
<span class="n">teardown</span><span class="o">=</span><span class="n">teardown</span><span class="p">,</span>
337319
<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>
339320
<span class="n">cleanup_seconds</span><span class="o">=</span><span class="n">cleanup_seconds</span><span class="p">,</span>
340321
<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>
342322
<span class="p">)</span>
343323
<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>
344324

@@ -365,30 +345,21 @@ <h1>Source code for easy_slurm.jobs</h1><div class="highlight"><pre>
365345
<span class="n">setup_resume</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
366346
<span class="n">teardown</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
367347
<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>
369348
<span class="n">cleanup_seconds</span><span class="p">:</span> <span class="nb">int</span><span class="p">,</span>
370349
<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>
372350
<span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">str</span><span class="p">:</span>
373351
<span class="w"> </span><span class="sd">&quot;&quot;&quot;Returns source for job script.&quot;&quot;&quot;</span>
374352
<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>
376353

377354
<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>
378355
<span class="n">easy_slurm_version</span><span class="o">=</span><span class="n">__version__</span><span class="p">,</span>
379356
<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>
381357
<span class="n">resubmit_limit</span><span class="o">=</span><span class="n">resubmit_limit</span><span class="p">,</span>
382358
<span class="p">)</span>
383359

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-
387360
<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>
388361
<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>
389362
<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>
392363

393364
<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>
394365
<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>
406377
<span class="n">setup</span><span class="o">=</span><span class="n">setup</span><span class="p">,</span>
407378
<span class="n">setup_resume</span><span class="o">=</span><span class="n">setup_resume</span><span class="p">,</span>
408379
<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>
411380
<span class="p">)</span></div>
412381

413382

_sources/index.rst.txt

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,6 @@ To submit a job, simply fill in the various parameters shown in the example belo
4242
job_dir="$HOME/jobs/{date}-{job_name}",
4343
src="./src",
4444
assets="./assets",
45-
dataset="./data.tar.gz",
4645
setup="""
4746
virtualenv "$SLURM_TMPDIR/env"
4847
source "$SLURM_TMPDIR/env/bin/activate"
@@ -66,7 +65,7 @@ To submit a job, simply fill in the various parameters shown in the example belo
6665
resubmit_limit=64, # Automatic resubmission limit.
6766
)
6867
69-
All job files will be kept in the ``job_dir`` directory. Provide directory paths to ``src`` and ``assets`` -- these will be archived and copied to the ``job_dir`` directory. Provide a file path to an archive containing the ``dataset``. Also provide Bash code in the hooks, which will be run in the following order:
68+
All job files will be kept in the ``job_dir`` directory. Provide directory paths to ``src`` and ``assets`` -- these will be archived and copied to the ``job_dir`` directory. Also provide Bash code in the hooks, which will be run in the following order:
7069

7170
.. list-table:: Hooks order
7271
:widths: 50 50
@@ -98,7 +97,6 @@ Jobs can also be fully configured using YAML files. See `examples/simple_yaml`_.
9897
job_dir: "$HOME/jobs/{date}-{job_name}"
9998
src: "./src"
10099
assets: "./assets"
101-
dataset: "./data.tar.gz"
102100
setup: |
103101
virtualenv "$SLURM_TMPDIR/env"
104102
source "$SLURM_TMPDIR/env/bin/activate"

0 commit comments

Comments
 (0)