Skip to content

Commit b287b73

Browse files
committed
refactor: Simplify function arguments and update progress bar parameter in controls.py
1 parent 6f30df5 commit b287b73

2 files changed

Lines changed: 6 additions & 2 deletions

File tree

brainpy/_src/math/object_transform/controls.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -308,12 +308,12 @@ def scan(
308308
The stacked outputs of ``body_fun`` when scanned over the leading axis of the inputs.
309309
"""
310310
return brainstate.transform.scan(
311-
body_fun=body_fun,
311+
body_fun,
312312
init=init,
313313
xs=operands,
314314
reverse=reverse,
315315
unroll=unroll,
316-
progress_bar=brainstate.transform.ProgressBar() if progress_bar else None,
316+
pbar=brainstate.transform.ProgressBar() if progress_bar else None,
317317
)
318318

319319

brainpy/_src/math/random.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -551,6 +551,10 @@ def value(self):
551551
record_state_value_read(self)
552552
return self._read_value()
553553

554+
# def check_if_deleted(self):
555+
# if self.value.is_deleted():
556+
# self.seed()
557+
554558
def split_key(self):
555559
"""Create a new seed from the current seed.
556560
"""

0 commit comments

Comments
 (0)