oh-my-batch 0.4.7__py3-none-any.whl → 0.4.8__py3-none-any.whl

oh_my_batch/job.py

@@ -94,6 +94,25 @@ class BaseJobManager:
                 error = True
         if error:
             raise RuntimeError('Some jobs failed')
+    
+    def wait(self, *job_ids, timeout=None, interval=10):
+        """
+        Wait for jobs to finish
+
+        :param job_ids: Job ids to wait for
+        :param timeout: Timeout in seconds
+        :param interval: Interval in seconds for checking job status
+        """
+        current = time.time()
+        while True:
+            jobs = [{'id': j, 'state': JobState.NULL} for j in job_ids]
+            jobs = self._update_state(jobs)
+            if all(JobState.is_terminal(j['state']) for j in jobs):
+                break
+            if timeout and time.time() - current > timeout:
+                logger.error('Timeout, current state: %s', jobs)
+                break
+            time.sleep(interval)
 
     def _update_jobs(self, jobs: List[dict], max_tries: int, submit_opts: str):
         jobs = self._update_state(jobs)
@@ -135,7 +154,7 @@ class Slurm(BaseJobManager):
                 logger.error('Failed to query job status: %s', log_cp(cp))
                 return jobs
             out = cp.stdout.decode('utf-8')
-            logger.info('Job status:\n%s', out)
+            logger.debug('Job status:\n%s', out)
             new_state = parse_csv(out)
         else:
             new_state = []

{oh_my_batch-0.4.7.dist-info → oh_my_batch-0.4.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oh-my-batch
-Version: 0.4.7
+Version: 0.4.8
 Summary: 
 License: GPL
 Author: weihong.xu
@@ -44,16 +44,6 @@ To make the best use of `oh-my-batch`, you need to know some shell tips.
 * [Run multiple line shell script with ssh](https://stackoverflow.com/a/32082912/3099733)
 
 ## Use cases
-### Load functions in shell script
-You can load useful functions from `oh-my-batch` this way:
-
-```bash
-eval "$(omb misc export-shell-func)"
-# or
-omb misc export-shell-func > omb-func.sh && source omb-func.sh
-```
-
-This will load extra functions to your shell script, for example, `checkpoint`.
 
 ### Generate files from different combinations of parameters
 

{oh_my_batch-0.4.7.dist-info → oh_my_batch-0.4.8.dist-info}/RECORD

@@ -5,11 +5,11 @@ oh_my_batch/assets/functions.sh,sha256=LaiavZBu84D7C1r-dWhf91vPTuHXCMV1DQZUIPVQn
 oh_my_batch/batch.py,sha256=o4G7AlaBOUNtwCWfO5bGnPA-iWhhq_iJxpALy1oIYb8,3593
 oh_my_batch/cli.py,sha256=Jyz8q2pUYke3mfJS6F_G9S9hApddgXxQw1BsN6Kfkjc,553
 oh_my_batch/combo.py,sha256=A5flYsjugcelj105P6Nq4M3kNGtdqh3c7QaSIL3hUcg,9954
-oh_my_batch/job.py,sha256=Foq-xyTj01KaCxZXmfXqTZ8orFL9ZVLix2dqT5w-B3Q,6515
+oh_my_batch/job.py,sha256=8qFYscHGzIQhX-HrAGxCCv7h2T8Ls1Fj0gmNEBdP5Go,7227
 oh_my_batch/misc.py,sha256=G_iOovRCrShBJJCc82QLN0CvMqW4adOefEoY1GedEiw,452
 oh_my_batch/util.py,sha256=5ve2QcviuF0UHFLrsXmjMTj0ogXJ4g05q1y-yWCFuOk,2409
-oh_my_batch-0.4.7.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-oh_my_batch-0.4.7.dist-info/METADATA,sha256=IbPGeI3srsM97vpX15osZ-dgb3s3pNLqIxsK9IdikT0,5384
-oh_my_batch-0.4.7.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-oh_my_batch-0.4.7.dist-info/entry_points.txt,sha256=ZY2GutSoNjjSyJ4qO2pTeseKUFgoTYdvmgkuZZkwi68,77
-oh_my_batch-0.4.7.dist-info/RECORD,,
+oh_my_batch-0.4.8.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+oh_my_batch-0.4.8.dist-info/METADATA,sha256=dXaGpQ8Dx7RgtLfdeuZ84uhLw3ZBw43k1FE2VXpY-og,5091
+oh_my_batch-0.4.8.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+oh_my_batch-0.4.8.dist-info/entry_points.txt,sha256=ZY2GutSoNjjSyJ4qO2pTeseKUFgoTYdvmgkuZZkwi68,77
+oh_my_batch-0.4.8.dist-info/RECORD,,