Merge pull request #1040 from joelgarde:joelgarde-patch-1

PiperOrigin-RevId: 359516268
google · Feb 25, 2021 · 7bc3d5c · 7bc3d5c
2 parents 9ae3e18 + 1749800
commit 7bc3d5c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flax/linen/module.py b/flax/linen/module.py
@@ -368,7 +368,7 @@ class Module:
     from flax import linen as nn
 
     class Module(nn.Module):
-      features: Tuple[int] = [16, 4]
+      features: Tuple[int] = (16, 4)
 
       def setup(self):
         self.dense1 = Dense(self.features[0])