tuple

eringrant · Dec 2, 2023 · 2d8622f · 2d8622f
1 parent 1cd8cc4
commit 2d8622f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nets/models/feedforward.py b/nets/models/feedforward.py
@@ -108,7 +108,7 @@ def __init__(
 
     for key, ins, outs, drop in zip(
       hidden_keys,
-      (in_features,) + hidden_features[:-1],
+      (in_features, *tuple(hidden_features[:-1])),
       hidden_features,
       dropout_probs[:-1],
       strict=True,