Revert "Add code for processing and computing with progress bar"

This reverts commit 12977af.
brainpy · Oct 25, 2024 · b8e04d8 · b8e04d8
1 parent 12977af
commit b8e04d8
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 20 deletions.
diff --git a/brainpy/_src/integrators/runner.py b/brainpy/_src/integrators/runner.py
@@ -318,11 +318,8 @@ def run(
     hists = self._run_fun_integration(args, dyn_args, times, indices)
     if eval_time:
       running_time = time.time() - t0
-
-    # due to jax 0.4.32 enable the async dispatch(https://github.com/jax-ml/jax/blob/main/CHANGELOG.md#jax-0432-september-11-2024),
-    # we temporarily do not close the progress bar
-    # if self.progress_bar:
-    #   self._pbar.close()
+    if self.progress_bar:
+      self._pbar.close()
 
     # post-running
     times += self.dt

diff --git a/brainpy/_src/runners.py b/brainpy/_src/runners.py
@@ -486,11 +486,8 @@ def predict(
       running_time = time.time() - t0
 
     # close the progress bar
-
-    # due to jax 0.4.32 enable the async dispatch(https://github.com/jax-ml/jax/blob/main/CHANGELOG.md#jax-0432-september-11-2024),
-    # we temporarily do not close the progress bar
-    # if self.progress_bar:
-    #   self._pbar.close()
+    if self.progress_bar:
+      self._pbar.close()
 
     # post-running for monitors
     if self._memory_efficient:

diff --git a/brainpy/_src/train/offline.py b/brainpy/_src/train/offline.py
@@ -193,11 +193,8 @@ def fit(
     del monitor_data
 
     # close the progress bar
-
-    # due to jax 0.4.32 enable the async dispatch(https://github.com/jax-ml/jax/blob/main/CHANGELOG.md#jax-0432-september-11-2024),
-    # we temporarily do not close the progress bar
-    # if self.progress_bar:
-    #   self._pbar.close()
+    if self.progress_bar:
+      self._pbar.close()
 
     # final things
     for node in self.train_nodes:

diff --git a/brainpy/_src/train/online.py b/brainpy/_src/train/online.py
@@ -191,11 +191,8 @@ def fit(
     outs, hists = self._fit(indices, xs=xs, ys=ys, shared_args=shared_args)
 
     # close the progress bar
-
-    # due to jax 0.4.32 enable the async dispatch(https://github.com/jax-ml/jax/blob/main/CHANGELOG.md#jax-0432-september-11-2024),
-    # we temporarily do not close the progress bar
-    # if self.progress_bar:
-    #   self._pbar.close()
+    if self.progress_bar:
+      self._pbar.close()
 
     # post-running for monitors
     if self.numpy_mon_after_run: