resubmit 0.0.7__py3-none-any.whl → 0.0.9__py3-none-any.whl

resubmit/__bookkeeping.py

@@ -153,6 +153,7 @@ def submit_jobs(
     block: bool = False,
     prompt: bool = True,
     local_run: bool = False,
+    debug: bool = False,
     job_name: Optional[str] = "resubmit",
     slurm_additional_parameters: Dict | None = None,
 ) -> Any:
@@ -174,6 +175,8 @@ def submit_jobs(
         block: Whether to block until jobs complete.
         prompt: Whether to prompt for confirmation before submission.
         local_run: If True, runs the function locally instead of submitting.
+        debug: If True, it runs only the first job in the queue for debugging.
+        job_name: Name of the job.
         slurm_additional_parameters: Additional Slurm parameters as a dict. If not provided, defaults to {"gpus": num_gpus}.
     Returns:
         The result of `submit_jobs` from `.__submit`.
@@ -194,6 +197,7 @@ def submit_jobs(
         block=block,
         prompt=prompt,
         local_run=local_run,
+        debug=debug,
         job_name=job_name,
         slurm_additional_parameters=slurm_additional_parameters,
     )

resubmit/__submit.py

@@ -15,12 +15,28 @@ def _submit_jobs(
     block: bool,
     prompt: bool,
     local_run: bool,
+    debug: bool = False,
     job_name: Optional[str] = "resubmit",
     slurm_additional_parameters: Optional[Dict] = None,
 ):
     """Submit jobs described by `jobs_args` where each entry is a dict of kwargs for `func`.
 
-    - If `local_run` is True, the function is called directly: `func(jobs_args)`.
+    Args:
+        jobs_args: Iterable of dicts of job parameters.
+        func: Function to be submitted for each job.
+        timeout_min: Job timeout in minutes.
+        cpus_per_task: Number of CPUs per task.
+        mem_gb: Memory in GB.
+        num_gpus: Number of GPUs.
+        folder: Folder for logs.
+        block: Whether to block until jobs complete.
+        prompt: Whether to prompt for confirmation before submission.
+        local_run: If True, runs the function locally instead of submitting.
+        debug: If True, runs only the first job in the queue for debugging.
+        job_name: Name of the job.
+        slurm_additional_parameters: Additional Slurm parameters as a dict. If not provided,
+    
+    - If `local_run` is True, the function is called directly on the local machine: `func(jobs_args[0])`.
     - Otherwise, submits via submitit.AutoExecutor and returns job objects or, if `block` is True, waits and returns results.
 
     Optional Slurm settings `constraint` and `reservation` can be provided via explicit
@@ -30,12 +46,16 @@ def _submit_jobs(
     """
     jobs_list = list(jobs_args) if not isinstance(jobs_args, list) else jobs_args
 
+    if debug:
+        print("Debug mode: only running the first job locally")
+        return func(jobs_list[0])
+
     if len(jobs_list) == 0:
         print("No jobs to run exiting")
         return
 
     if local_run:
-        print("Running locally (local_run=True)")
+        print("Running the jobs locally (local_run=True)")
         return func(jobs_list)
 
     if prompt:

{resubmit-0.0.7.dist-info → resubmit-0.0.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: resubmit
-Version: 0.0.7
+Version: 0.0.9
 Summary: Small wrapper around submitit to simplify cluster submissions
 Author: Amir Mehrpanah
 License: MIT

resubmit-0.0.9.dist-info/RECORD

@@ -0,0 +1,9 @@
+resubmit/__bookkeeping.py,sha256=5BurUVe7VHJWnU_rT7e0U04d7hybR6YSYZUbOexk3AM,7448
+resubmit/__debug.py,sha256=8RINyz7eSAiT47d018wR0R3B_u4PllQJCiLy0zTSQDE,887
+resubmit/__init__.py,sha256=FLKq6KZeI973gBXzdnSkvK1aEdxF--5V2T82fxyzv0U,219
+resubmit/__submit.py,sha256=_UR1vrmHK5H2G-R0_nm7G7Xy3RW2R1DXSyrrRZ2081c,3038
+resubmit-0.0.9.dist-info/licenses/LICENSE,sha256=v2spsd7N1pKFFh2G8wGP_45iwe5S0DYiJzG4im8Rupc,1066
+resubmit-0.0.9.dist-info/METADATA,sha256=SFZ6WuiTjgI3uWg2-mXx5j1snX_MzZD14OEQR2OVwOM,2976
+resubmit-0.0.9.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+resubmit-0.0.9.dist-info/top_level.txt,sha256=BfCexfX-VhUZuNi8sI88i0HF_e3ppausQ76hxPeXjYc,9
+resubmit-0.0.9.dist-info/RECORD,,

resubmit-0.0.7.dist-info/RECORD

@@ -1,9 +0,0 @@
-resubmit/__bookkeeping.py,sha256=6Q6IvstSCQH75WY430AbtOWFHWu9lVDnaJD4RJuCLrA,7288
-resubmit/__debug.py,sha256=8RINyz7eSAiT47d018wR0R3B_u4PllQJCiLy0zTSQDE,887
-resubmit/__init__.py,sha256=FLKq6KZeI973gBXzdnSkvK1aEdxF--5V2T82fxyzv0U,219
-resubmit/__submit.py,sha256=SgtZFb-Yi7zIv9BGVyXgbsjca_s6_kZlA9mBO9ZWGtg,2149
-resubmit-0.0.7.dist-info/licenses/LICENSE,sha256=v2spsd7N1pKFFh2G8wGP_45iwe5S0DYiJzG4im8Rupc,1066
-resubmit-0.0.7.dist-info/METADATA,sha256=IGWx0_wsZe0s4POwQnhK_SsA2SzEzRZxPNoQNaDNFYk,2976
-resubmit-0.0.7.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-resubmit-0.0.7.dist-info/top_level.txt,sha256=BfCexfX-VhUZuNi8sI88i0HF_e3ppausQ76hxPeXjYc,9
-resubmit-0.0.7.dist-info/RECORD,,