Skip to content

Commit e8b2938

Browse files
committed
Update version to 2.7.1 and refactor Dynamics class: enhance input handling and add new properties
1 parent cd22e00 commit e8b2938

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

brainpy/state/_base.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -405,17 +405,17 @@ def align_pre(self, dyn: Union[ParamDescriber[T], T]) -> T:
405405
>>> n1.align_pre(brainstate.nn.Expon.desc(n1.varshape)) # n2 will run after n1
406406
"""
407407
if isinstance(dyn, Dynamics):
408-
self._add_after_update(id(dyn), dyn)
408+
self.add_after_update(id(dyn), dyn)
409409
return dyn
410410
elif isinstance(dyn, ParamDescriber):
411411
if not issubclass(dyn.cls, Dynamics):
412412
raise TypeError(f'The input {dyn} should be an instance of {Dynamics}.')
413-
if not self._has_after_update(dyn.identifier):
414-
self._add_after_update(
413+
if not self.has_after_update(dyn.identifier):
414+
self.add_after_update(
415415
dyn.identifier,
416416
dyn() if ('in_size' in dyn.kwargs or len(dyn.args) > 0) else dyn(in_size=self.varshape)
417417
)
418-
return self._get_after_update(dyn.identifier)
418+
return self.get_after_update(dyn.identifier)
419419
else:
420420
raise TypeError(f'The input {dyn} should be an instance of {Dynamics} or a delayed initializer.')
421421

0 commit comments

Comments
 (0)