make predict a little bit faster (aimclub#83)

jrzkaminski · Oct 25, 2023 · 13742c5 · 13742c5
1 parent 56c9c58
commit 13742c5
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/bamt/networks/base.py b/bamt/networks/base.py
@@ -684,7 +684,12 @@ def wrapper():
                         output[node.name] = node.choose(node_data, pvals=pvals)
             return output
 
-        if progress_bar:
+        if predict:
+            seq = []
+            for _ in tqdm(range(n), position=0, leave=True):
+                result = wrapper()
+                seq.append(result)
+        elif progress_bar:
             seq = Parallel(n_jobs=parall_count)(
                 delayed(wrapper)() for _ in tqdm(range(n), position=0, leave=True)
             )