From 13742c58025355f9f902dda9640eb7d09ebac5bf Mon Sep 17 00:00:00 2001
From: Kovalev Nikita <44478476+Yakonick@users.noreply.github.com>
Date: Wed, 25 Oct 2023 16:39:39 +0300
Subject: [PATCH] make predict a little bit faster (#83)

---
 bamt/networks/base.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/bamt/networks/base.py b/bamt/networks/base.py
index b890728..09ddc7d 100644
--- 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)
             )