Skip to content

Commit 64fffb0

Browse files
committed
updated readme
1 parent c997093 commit 64fffb0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

flax/experimental/nnx/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ NNX Modules are normal python classes, they obey regular python semantics such a
142142

143143
```python
144144
class Foo(nnx.Module):
145-
def __init__(self, rngs: nnx.Rngs):
145+
def __init__(self, *, rngs: nnx.Rngs):
146146
# node attributes
147147
self.param = nnx.Param(jnp.array(1))
148148
self.submodule = nnx.Linear(12, 3, rngs=rngs)

0 commit comments

Comments
 (0)