Skip to content

Commit ca7d748

Browse files
author
Flax Authors
committed
Merge pull request #3563 from chiamp:readme
PiperOrigin-RevId: 591940459
2 parents a59675c + 64fffb0 commit ca7d748

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)