Update

[ghstack-poisoned]

Author: fegin

pyproject.toml

@@ -26,7 +26,8 @@ dev = [
     "black",
     "pyre-check",
     "parameterized",
-    "expecttest"
+    "expecttest",
+    "numpy"
 ]
 
 [tool.maturin]

torchft/fsdp_test.py

@@ -47,7 +47,6 @@ def setUp(self) -> None:
 
     def test_fsdp(self) -> None:
         group_size = self.world_size // 2
-        # pyre-ignore[16]
         group = self.rank // group_size
         group_rank = self.rank % group_size
 

torchft/process_group_test.py

@@ -316,7 +316,6 @@ def setUp(self) -> None:
 
     def test_init_device_mesh(self) -> None:
         os.environ["MASTER_PORT"] = str(12346)
-        # pyre-ignore[16]
         os.environ["RANK"] = str(self.rank)
         os.environ["WORLD_SIZE"] = str(4)
 
@@ -332,7 +331,6 @@ def test_init_device_mesh(self) -> None:
             manager=manager,
         )
 
-        # pyre-ignore[16]
         self.assertTrue(
             isinstance(device_mesh.get_group("dp_replicate"), ManagedProcessGroup)
         )