slurm-api: Allow requesting for a specific partition.

* ravanan/batch-system.scm (<slurm-api-batch-system>)[partition]: Add
field.
* bin/ravanan (%options): Add slurm-partition.
(print-usage): Document it.
(main): Initialize partition field of <slurm-api-batch-system> object.
* ravanan/command-line-tool.scm (run-command-line-tool): Pass
#:partition argument to submit-job.
* ravanan/slurm-api.scm (submit-job): Accept #:partition argument.

1 files changed, 6 insertions, 1 deletions

diff --git a/bin/ravanan b/bin/ravanan
index 5807a8e..2bcf0db 100755
--- a/bin/ravanan
+++ b/bin/ravanan
@@ -57,6 +57,9 @@ exec guile --no-auto-compile -e main -s "$0" "$@"
         (option (list "slurm-jwt") #t #f
                 (lambda (opt name arg result)
                   (acons 'slurm-jwt arg result)))
+        (option (list "slurm-partition") #t #f
+                (lambda (opt name arg result)
+                  (acons 'slurm-partition arg result)))
         (option (list "help") #f #t
                 (lambda (opt name arg result)
                   (acons 'help #t result)))))
@@ -85,6 +88,7 @@ Slurm API batch system options:
 
   --slurm-api-endpoint=SLURM-API-ENDPOINT    slurm API endpoint to connect to
   --slurm-jwt=SLURM-JWT                      slurm JWT to authenticate with
+  --slurm-partition=SLURM-PARTITION          slurm partition to request
 "
           program))
 
@@ -152,7 +156,8 @@ files that have the token in the @verbatim{SLURM_JWT=token} format."
                                       (slurm-api-batch-system
                                        (assq-ref args 'slurm-api-endpoint)
                                        (and (assq-ref args 'slurm-jwt)
-                                            (read-jwt (assq-ref args 'slurm-jwt))))))
+                                            (read-jwt (assq-ref args 'slurm-jwt)))
+                                       (assq-ref args 'slurm-partition))))
                                    #:guix-daemon-socket (assq-ref args 'guix-daemon-socket))
                      (current-output-port)
                      #:pretty #t)