oh-my-batch 0.4.5__tar.gz → 0.4.7__tar.gz

{oh_my_batch-0.4.5 → oh_my_batch-0.4.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oh-my-batch
-Version: 0.4.5
+Version: 0.4.7
 Summary: 
 License: GPL
 Author: weihong.xu
@@ -13,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: fire (>=0.5.0,<0.6.0)
 Description-Content-Type: text/markdown
 
@@ -120,15 +121,12 @@ EOF
 omb batch \
     add_work_dirs tmp/tasks/* - \
     add_header_files tmp/lammps_header.sh - \
-    add_cmds "checkpoint lmp.done ./run.sh" - \
+    add_cmds "./run.sh" - \
     make tmp/lmp-{i}.slurm --concurrency 2
 ```
 
 You will find batch scripts `tmp/lmp-0.slurm` and `tmp/lmp-1.slurm` in `tmp` directory.
 
-`omb batch` will provide some useful functions in the batch script.
-For example, `checkpoint` will check if the job is done and skip the job if it's done.
-
 You can run the above script by `./examples/omb-batch.sh`,
 
 ### Track the state of job in job schedular

{oh_my_batch-0.4.5 → oh_my_batch-0.4.7}/README.md

@@ -102,15 +102,12 @@ EOF
 omb batch \
     add_work_dirs tmp/tasks/* - \
     add_header_files tmp/lammps_header.sh - \
-    add_cmds "checkpoint lmp.done ./run.sh" - \
+    add_cmds "./run.sh" - \
     make tmp/lmp-{i}.slurm --concurrency 2
 ```
 
 You will find batch scripts `tmp/lmp-0.slurm` and `tmp/lmp-1.slurm` in `tmp` directory.
 
-`omb batch` will provide some useful functions in the batch script.
-For example, `checkpoint` will check if the job is done and skip the job if it's done.
-
 You can run the above script by `./examples/omb-batch.sh`,
 
 ### Track the state of job in job schedular

{oh_my_batch-0.4.5 → oh_my_batch-0.4.7}/oh_my_batch/batch.py

@@ -86,9 +86,6 @@ class BatchMaker:
         :param path: Path to save batch script files, use {i} to represent index
         :param concurrency: Number of scripts to to make
         """
-        # inject pre-defined functions
-        self.add_header_files(get_asset('functions.sh'))
-
         header = '\n'.join(self._script_header)
         bottom = '\n'.join(self._script_bottom)
         for i, work_dirs in enumerate(split_list(self._work_dirs, concurrency)):

{oh_my_batch-0.4.5 → oh_my_batch-0.4.7}/oh_my_batch/job.py

@@ -87,7 +87,13 @@ class BaseJobManager:
                 break
 
             time.sleep(interval)
-
+        for job in jobs:
+            error = False
+            if not JobState.is_success(job['state']):
+                logger.error('Job %s failed', job['script'])
+                error = True
+        if error:
+            raise RuntimeError('Some jobs failed')
 
     def _update_jobs(self, jobs: List[dict], max_tries: int, submit_opts: str):
         jobs = self._update_state(jobs)

{oh_my_batch-0.4.5 → oh_my_batch-0.4.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "oh-my-batch"
-version = "0.4.5"
+version = "0.4.7"
 description = ""
 authors = ["weihong.xu <xuweihong.cn@gmail.com>"]
 license = "GPL"