Make priority class a CLI argument, relocate k8s assets into package data subdirectory

pyproject.toml

@@ -16,6 +16,9 @@ jobs_execute = "jobs.execute:execute"
 [project.optional-dependencies]
 dev = ["black", "ruff", "ruff-lsp", "pytest","pre-commit"]
 
+[tool.setuptools.package-data]
+"jobs.assets" = ["*.yaml"]
+
 [tool.ruff]
 src = ["src"]
 

src/jobs/cli.py

@@ -44,6 +44,11 @@ def _make_argparser() -> argparse.ArgumentParser:
         help="Kubernetes namespace to create resources in, defaults to currently active namespace",
     )
 
+    parser.add_argument(
+        "--priority",
+        help="Job priority to set when submitting to Kueue (execution mode 'kueue' only).",
+    )
+
     return parser
 
 
@@ -66,6 +71,7 @@ def submit_job(job: Job) -> None:
         runner = KueueRunner(
             namespace=args.namespace,
             local_queue=args.kueue_local_queue,
+            priority_class=args.priority,
         )
         runner.run(job, image)
     elif mode == ExecutionMode.RAYCLUSTER:

src/jobs/runner/kueue.py

@@ -17,6 +17,7 @@ def __init__(self, **kwargs: str) -> None:
         super().__init__()
 
         self._queue = kwargs.get("local_queue", "user-queue")
+        self._priority_class = kwargs.get("priority_class", None)
 
     def _make_job_crd(self, job: Job, image: Image, namespace: str) -> client.V1Job:
         def _assert_kueue_localqueue(name: str) -> bool:
@@ -51,11 +52,11 @@ def _assert_kueue_workloadpriorityclass(name: str) -> bool:
                     raise ValueError(
                         f"Specified Kueue local queue does not exist: {queue!r}"
                     )
-            if pc := sched_opts.priority_class:
-                if not _assert_kueue_workloadpriorityclass(pc):
-                    raise ValueError(
-                        f"Specified Kueue workload priority class does not exist: {pc!r}"
-                    )
+        if pc := self._priority_class:
+            if not _assert_kueue_workloadpriorityclass(pc):
+                raise ValueError(
+                    f"Specified Kueue workload priority class does not exist: {pc!r}"
+                )
 
         metadata = client.V1ObjectMeta(
             generate_name=sanitize_rfc1123_domain_name(job.name),
@@ -64,9 +65,7 @@ def _assert_kueue_workloadpriorityclass(name: str) -> bool:
                     "kueue.x-k8s.io/queue-name": sched_opts.queue_name
                     if sched_opts and sched_opts.queue_name
                     else self._queue,
-                    "kueue.x-k8s.io/priority-class": sched_opts.priority_class
-                    if sched_opts
-                    else None,
+                    "kueue.x-k8s.io/priority-class": self._priority_class or None,
                 }
             ),
         )