Fix Pyre

Summary: not sure why old pyre ignores are erroring - probably pyre update that changes what styles are accepted

Differential Revision: D74817490

Author: aporialiao

torchrec/optim/tests/test_keyed.py

@@ -81,7 +81,7 @@ def test_load_state_dict(self) -> None:
                     "one": 1.0,
                     "tensor": torch.tensor([5.0, 6.0]),
                     "sharded_tensor": sharded_tensor.full(
-                        # pyre-ignore [28]
+                        # pyre-fixme
                         sharding_spec.ChunkShardingSpec(
                             dim=0, placements=["rank:0/cpu"]
                         ),
@@ -116,7 +116,7 @@ def test_load_state_dict(self) -> None:
                 "one": 1.0,
                 "tensor": torch.tensor([5.0, 6.0]),
                 "sharded_tensor": sharded_tensor.full(
-                    # pyre-ignore [28]
+                    # pyre-fixme
                     sharding_spec.ChunkShardingSpec(dim=0, placements=["rank:0/cpu"]),
                     (4,),
                     fill_value=1.0,
@@ -157,7 +157,7 @@ def test_load_state_dict(self) -> None:
         expected_state_dict["state"]["param_1"]["tensor"] = torch.tensor([50.0, 60.0])
         # pyre-ignore [6]
         expected_state_dict["state"]["param_1"]["sharded_tensor"] = sharded_tensor.full(
-            # pyre-ignore [28]
+            # pyre-fixme
             sharding_spec.ChunkShardingSpec(dim=0, placements=["rank:0/cpu"]),
             (4,),
             fill_value=10.0,