From 9024770ce8a6050ee570343233bee7dd47d37814 Mon Sep 17 00:00:00 2001 From: Aareon Sullivan Date: Fri, 22 Mar 2024 19:16:58 -0500 Subject: [PATCH] Change subfunction signature to ignore `value` Instead of marking `value` for deletion by doing ```py del value # Unused. ``` ignore the variable in the function signature. --- model.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/model.py b/model.py index 87d700d..ff08762 100644 --- a/model.py +++ b/model.py @@ -90,9 +90,7 @@ def ffn_size(emb_size, widening_factor): def apply_rules(rules): - def _apply_rules(path, value): - del value # Unused. - + def _apply_rules(path, _): path_list = [str(i.key).split("/") for i in path if isinstance(i, jax.tree_util.DictKey)] flattened_path = jax.tree_util.tree_flatten(path_list)[0]