From e406e63eadb57e5bc8be9cddd241beaf2972b6bc Mon Sep 17 00:00:00 2001
From: Lukas Winkler <git@lw1.at>
Date: Wed, 21 Dec 2022 15:47:28 +0100
Subject: [PATCH] allow setting qos and account

---
 jobrun/jobrun.py       | 4 ++++
 spectra_computation.py | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/jobrun/jobrun.py b/jobrun/jobrun.py
index 352b726..e894e4b 100644
--- a/jobrun/jobrun.py
+++ b/jobrun/jobrun.py
@@ -16,6 +16,8 @@ def jobrun(
         tasks: int = None,
         nodes: int = None,
         mem: int = None,
+        qos: str = None,
+        account: str = None,
         source: Path = None,
         sbatch: bool = True
 ):
@@ -32,6 +34,8 @@ def jobrun(
         "nodes": nodes,
         "tasks": tasks,
         "mem": mem,
+        "qos": qos,
+        "account": account,
         "job-name": name,
         "output": jobscript_dir / f"out-{name}-%a_%a.txt"
     }
diff --git a/spectra_computation.py b/spectra_computation.py
index 4b69993..af537e8 100644
--- a/spectra_computation.py
+++ b/spectra_computation.py
@@ -16,7 +16,7 @@ vsc = True
 
 def spectra_jobrun(args):
     if vsc:
-        jobrun(args, time="12:00:00", tasks=128, mem=128, source=Path("/gpfs/data/fs71636/lwinkler/spack_notes.txt"))
+        jobrun(args, time="12:00:00", nodes=1, source=Path("/gpfs/data/fs71636/lwinkler/spack-latest.sh"))
     else:
         subprocess.run(args, check=True)