Skip to content

Commit 951d2e0

Browse files
authored
Add files via upload
1 parent cb73981 commit 951d2e0

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

multioptpy/Wrapper/mapper.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2616,7 +2616,9 @@ def _run_batch_parallel(
26162616

26172617
# ── Step 1: cancel not-yet-started futures ────────────────
26182618
for future in futures_map:
2619-
future.cancel()
2619+
if future.cancel():
2620+
task, run_dir, iteration, gamma_sign, atom_i, atom_j = futures_map[future]
2621+
self.queue.release((task.node_id, tuple(task.afir_params)))
26202622

26212623
# ── Step 2: force-kill running worker processes ───────────
26222624
# executor._processes is a {pid: multiprocessing.Process} dict
@@ -3487,8 +3489,7 @@ def _enqueue_perturbations(self, node: EQNode, force_add: bool = False) -> None:
34873489
"graph has only 1 node — exclusion suppressed.",
34883490
node.node_id,
34893491
)
3490-
elif (node.node_id in self.excluded_node_ids
3491-
and self._node_has_been_explored(node.node_id)):
3492+
elif node.node_id in self.excluded_node_ids:
34923493
logger.debug(
34933494
"_enqueue_perturbations: EQ%d is in excluded_node_ids and has "
34943495
"already been explored at least once — skipped.",

0 commit comments

Comments
 (0)