lsst-ctrl-execute 29.2025.1400__tar.gz → 29.2025.2100__tar.gz

{lsst_ctrl_execute-29.2025.1400 → lsst_ctrl_execute-29.2025.2100}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lsst-ctrl-execute
-Version: 29.2025.1400
+Version: 29.2025.2100
 Summary: Utilities for executing and managing workloads.
 Project-URL: Homepage, https://github.com/lsst/ctrl_execute
 Author-email: Rubin Observatory Data Management <dm-admin@lists.lsst.org>

{lsst_ctrl_execute-29.2025.1400 → lsst_ctrl_execute-29.2025.2100}/python/lsst/ctrl/execute/allocatorParser.py

@@ -178,6 +178,24 @@ class AllocatorParser:
             dest="packnodes",
             help="encourage nodes to pack jobs rather than spread",
         )
+        parser.add_argument(
+            "--exclude",
+            action="store",
+            default=None,
+            dest="exclude",
+            help="machine name(s) to exclude Slurm jobs from",
+            type=str,
+            required=False,
+        )
+        parser.add_argument(
+            "--nodelist",
+            action="store",
+            default=None,
+            dest="nodelist",
+            help="machine name(s) to target Slurm jobs toward",
+            type=str,
+            required=False,
+        )
         parser.add_argument(
             "--exclusive",
             action="store_true",

{lsst_ctrl_execute-29.2025.1400 → lsst_ctrl_execute-29.2025.2100}/python/lsst/ctrl/execute/slurmPlugin.py

@@ -214,6 +214,16 @@ class SlurmPlugin(Allocator):
         else:
             self.defaults["RESERVATION"] = ""
 
+        if self.opts.exclude is not None:
+            self.defaults["EXCLUDE"] = f"#SBATCH --exclude {self.opts.exclude}"
+        else:
+            self.defaults["EXCLUDE"] = ""
+
+        if self.opts.nodelist is not None:
+            self.defaults["NODELIST"] = f"#SBATCH --nodelist {self.opts.nodelist}"
+        else:
+            self.defaults["NODELIST"] = ""
+
         if self.opts.exclusive is not None:
             self.defaults["EXCLUSIVE"] = "#SBATCH --exclusive"
         else:

{lsst_ctrl_execute-29.2025.1400 → lsst_ctrl_execute-29.2025.2100}/tests/test_allocatorParser.py

@@ -42,6 +42,10 @@ class TestAllocatorParser(lsst.utils.tests.TestCase):
             "12",
             "-m",
             "00:30:00",
+            "--exclude",
+            "sdfmilan003",
+            "--nodelist",
+            "sdfmilan004",
             "-q",
             "normal",
             "-O",
@@ -57,6 +61,8 @@ class TestAllocatorParser(lsst.utils.tests.TestCase):
         self.assertEqual(args.nodeCount, 64)
         self.assertEqual(args.cpus, 12)
         self.assertEqual(args.maximumWallClock, "00:30:00")
+        self.assertEqual(args.exclude, "sdfmilan003")
+        self.assertEqual(args.nodelist, "sdfmilan004")
         self.assertEqual(args.queue, "normal")
         self.assertEqual(args.outputLog, "outlog")
         self.assertEqual(args.errorLog, "errlog")