hotfix prev commit

dbbs-lab · Oct 18, 2023 · 48e29eb · 48e29eb
1 parent 1f5b873
commit 48e29eb
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/bsb/placement/strategy.py b/bsb/placement/strategy.py
@@ -206,8 +206,10 @@ def queue(self, pool, chunk_size):
     def place(self, chunk, indicators):
         for indicator in indicators.values():
             cell_type = indicator.cell_type
-            # Pass the voxelset if it exists
-            voxels = getattr(p, "voxelset", None)
             # Guess total number, not chunk number, as entities bypass chunking.
-            n = sum(np.sum(indicator.guess(voxels=voxels)) for p in self.partitions)
+            n = sum(
+                # Pass the voxelset if it exists
+                np.sum(indicator.guess(voxels=getattr(p, "voxelset", None)))
+                for p in self.partitions
+            )
             self.scaffold.create_entities(cell_type, n)