Skip to content

Commit 7c67169

Browse files
author
Alexander Ororbia
committed
got bernoulli-cell rewritten and unit-tested
1 parent 21f43f8 commit 7c67169

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ngclearn/components/input_encoders/bernoulliCell.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,8 @@ class BernoulliCell(JaxComponent):
3232
"""
3333

3434
def __init__(self, name, n_units, batch_size=1, **kwargs):
35-
#super().__init__(name, **kwargs)
36-
super(JaxComponent, self).__init__(name, **kwargs)
35+
super().__init__(name, **kwargs)
36+
#super(BernoulliCell, self).__init__(name, **kwargs)
3737

3838
## Layer Size Setup
3939
self.batch_size = batch_size

0 commit comments

Comments
 (0)