Skip to content

Commit

Permalink
Fix core run functions for placement, after_placement and after_conne…
Browse files Browse the repository at this point in the history
…ctivity
  • Loading branch information
drodarie committed Jul 17, 2024
1 parent 0549b0e commit fe1bfcd
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions bsb/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -270,7 +270,7 @@ def run_placement(self, strategies=None, fail_fast=True, pipelines=True):
if pipelines:
self.run_pipelines()
if strategies is None:
strategies = [*self.placement.values()]
strategies = set(self.placement.values())
strategies = PlacementStrategy.sort_deps(strategies)
with self.create_job_pool(fail_fast=fail_fast) as pool:
if pool.is_main():
Expand Down Expand Up @@ -309,7 +309,7 @@ def run_after_placement(self, hooks=None, fail_fast=None, pipelines=True):
Run after placement hooks.
"""
if hooks is None:
hooks = self.after_placement
hooks = set(self.after_placement.values())
with self.create_job_pool(fail_fast) as pool:
if pool.is_main():
pool.schedule(hooks)
Expand All @@ -321,7 +321,7 @@ def run_after_connectivity(self, hooks=None, fail_fast=None, pipelines=True):
Run after placement hooks.
"""
if hooks is None:
hooks = self.after_placement
hooks = set(self.after_connectivity.values())
with self.create_job_pool(fail_fast) as pool:
if pool.is_main():
pool.schedule(hooks)
Expand Down

0 comments on commit fe1bfcd

Please sign in to comment.