Skip to content

Commit 7f3e7c8

Browse files
authored
Update gaussianErrorCell.py (#98)
added `and not isinstance(sigma, int)`
1 parent 4af85dc commit 7f3e7c8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ngclearn/components/neurons/graded/gaussianErrorCell.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def __init__(self, name, n_units, batch_size=1, sigma=1., shape=None, **kwargs):
4141
else:
4242
_shape = (batch_size, shape[0], shape[1], shape[2]) ## shape is 4D tensor
4343
sigma_shape = (1,1)
44-
if not isinstance(sigma, float):
44+
if not isinstance(sigma, float) and not isinstance(sigma, int):
4545
sigma_shape = jnp.array(sigma).shape
4646
self.sigma_shape = sigma_shape
4747
self.shape = shape

0 commit comments

Comments
 (0)