amend

pytorch · Feb 6, 2024 · 1337c41 · 1337c41
1 parent 326afdc
commit 1337c41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_functorch.py b/test/test_functorch.py
@@ -362,7 +362,7 @@ def zero_grad(p):
                 for p in params.flatten_keys().values()
             )
             assert params.requires_grad
-            params.apply_(zero_grad)
+            params.apply_(zero_grad, filter_empty=True)
             assert params.requires_grad
 
     def test_repopulate(self):