Skip to content

Commit d862baf

Browse files
committed
run linter
1 parent 57a7392 commit d862baf

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

bayesflow/workflows/basic_workflow.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -738,8 +738,14 @@ def fit_online(
738738
metric evolution over epochs.
739739
"""
740740

741+
import multiprocessing as mp
742+
741743
dataset = OnlineDataset(
742-
simulator=self.simulator, batch_size=batch_size, num_batches=num_batches_per_epoch, adapter=self.adapter
744+
simulator=self.simulator,
745+
batch_size=batch_size,
746+
num_batches=num_batches_per_epoch,
747+
adapter=self.adapter,
748+
workers=mp.cpu_count(),
743749
)
744750

745751
return self._fit(

0 commit comments

Comments
 (0)