matensemble 0.3.3__tar.gz → 0.3.4__tar.gz

{matensemble-0.3.3 → matensemble-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matensemble
-Version: 0.3.3
+Version: 0.3.4
 Summary: An adaptive and highly asynchronous ensemble simulation workflow manager MatEnsemble (https://github.com/Q-CAD/MatEnsemble) built jointly on top of the hierarchical graph based scheduler FLUX and concurrent-futures infrastructure of python
 Author: Soumendu Bagchi, Kaleb Duchesneau
 Author-email: Soumendu Bagchi <soumendubagchi@gmail.com>, Kaleb Duchesneau <kalebduchesneau@gmail.com>

{matensemble-0.3.3 → matensemble-0.3.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "matensemble"
-version = "0.3.3"
+version = "0.3.4"
 description = "An adaptive and highly asynchronous ensemble simulation workflow manager MatEnsemble (https://github.com/Q-CAD/MatEnsemble) built jointly on top of the hierarchical graph based scheduler FLUX and concurrent-futures infrastructure of python"
 readme = "README.md"
 license-files = ["LICENSE"]

{matensemble-0.3.3 → matensemble-0.3.4}/src/matensemble/TODO.md

@@ -171,6 +171,17 @@
 - [x] Convert Scaffold into PowerPoint Presentation
 - [x] Email coordinator about length of presentation and audience
 
+# === May 29, 2026 ===
+- [x] Fix the strategy error
+- [x] Make sure conda env is working
+- [ ] Make sure all containers are working and stable
+    * Frontier
+    * Perlmutter
+    * Pathfinder
+    * Linux
+- [ ] Update tutorials for all systems
+- [ ] Make all figures for JOSS paper and explain all of them
+
 ## === AFTER EVERYTHING ABOVE IS DONE AND STABLE ===
 
 ## --- Model Context Protocol ---
@@ -197,12 +208,12 @@
 
 ## --- Create first draft for JOSS ---
 - [ ] Read some example papers
-- [ ] Create draft and show Dr. Bagchi
-- [ ] Polish the repository to be ready for review
+- [x] Create draft and show Dr. Bagchi
+- [x] Polish the repository to be ready for review
 - [ ] Make sure that the tests work
 - [ ] Make sure that the example workflows work correctly
-- [ ] Make sure that they can easily test the code and
-- [ ] Create a conda package that they can easily test the code without having
+- [ ] Make sure that they can easily test the code
+- [x] Create a conda package that they can easily test the code without having
       to compile flux and flux-sched themselves
 
 ## --- Reading List ---

{matensemble-0.3.3 → matensemble-0.3.4}/src/matensemble/pipeline.py

@@ -278,7 +278,8 @@ class Pipeline:
         Returns
         -------
         Callable
-            A dummy function that just prints a warning to the stdout
+            A dummy function that just prints a warning to the stdout. The
+            actual function is stored in the registry
         """
 
         def decorator(func: Callable[..., Any]) -> Callable[..., Any]:
@@ -603,9 +604,6 @@ class Pipeline:
                 self._finished = True
             return self._collect_results()
 
-    def add_user_strat(self, chore_name: str, bolo_list: list[str]):
-        self._strategy_spec = {"name": chore_name, "bolo_list": bolo_list}
-
     def graph(self) -> nx.DiGraph:
         return self._create_graph()
 

{matensemble-0.3.3 → matensemble-0.3.4}/src/matensemble/strategy.py

@@ -224,9 +224,6 @@ class NonAdaptiveStrategy(FutureProcessingStrategy):
             return
 
 
-# TODO: Make the strategy look through the bolo list and spawn a new chore with the output of another
-#       and then make sure that the chore that is acting as a processing strat is a on the bolo list
-#       and if that is done then you need to get the results and spawn a new chore from the ChoreSpec
 class UserStrategy(FutureProcessingStrategy):
     def __init__(self, manager, pipeline, processing_chore, bolo_list) -> None:
         super().__init__(manager)
@@ -314,12 +311,13 @@ class UserStrategy(FutureProcessingStrategy):
                     # untrusted paths or third-party producers.
                     with (chore.workdir / "result.pickle").open("rb") as f:
                         chore_spec = pickle.load(f)
-                    new_chore, new_out = self.pipeline._spawn_chore_from_spec(
-                        chore_spec
-                    )
-                    self.pipeline._admit_spawned_chore(
-                        new_chore, new_out, self.manager
-                    )
+                    if chore_spec:
+                        new_chore, new_out = self.pipeline._spawn_chore_from_spec(
+                            chore_spec
+                        )
+                        self.pipeline._admit_spawned_chore(
+                            new_chore, new_out, self.manager
+                        )
                 except Exception as e:
                     self.manager._logger.exception(
                         f"FAILED TO SPAWN CHORE: chore={self.proc_chore} | due the following Exception ->\n{e}"