Update

[ghstack-poisoned]
pytorch · Jul 23, 2024 · 0cb0d40 · 0cb0d40
1 parent fc6a57d
commit 0cb0d40
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/test_cost.py b/test/test_cost.py
@@ -15083,10 +15083,10 @@ def __init__(self):
         assert v_p1 == v_p2
         assert v_params1 == v_params2
         assert v_buffers1 == v_buffers2
-        for p in mod.parameters():
-            assert isinstance(p, nn.Parameter)
-        for p in mod.buffers():
-            assert isinstance(p, Buffer)
+        for k, p in mod.named_parameters():
+            assert isinstance(p, nn.Parameter), k
+        for k, p in mod.named_buffers():
+            assert isinstance(p, Buffer), k
         for p in mod.actor_params.values(True, True):
             assert isinstance(p, (nn.Parameter, Buffer))
         for p in mod.value_params.values(True, True):