Remove default typing argument in PACKAGE file] [batch:20/511] [shard…

…:5/N]

Reviewed By: MaggieMoss

Differential Revision: D64659584

fbshipit-source-id: 3dcc9da20b3f52d652928ce3ace39327b1a7fc83

tests/test_memoryview_stream.py

@@ -17,7 +17,6 @@
 class MemoryviewStreamTest(unittest.TestCase):
     def test_memoryview_stream(self) -> None:
         tensor = torch.rand(1000)
-        # pyre-fixme[6]: For 1st argument expected `Buffer` but got `ndarray[Any, Any]`.
         mv = memoryview(tensor.numpy()).cast("b")
         self.assertEqual(len(mv), 4000)
 

tests/test_snapshot.py

@@ -180,7 +180,6 @@ def test_app_state_with_primitive_types(tmp_path: Path) -> None:
         bytes_key=None,
     )
 
-    # pyre-fixme[53]: Captured variable `snapshot` is not annotated.
     def _assert_primitive_entry_with_type(
         location_key: str, expected_type_name: str
     ) -> None:

torchsnapshot/memoryview_stream.py

@@ -13,13 +13,10 @@
 
 # pyre-fixme[13]: Attribute `write` is never initialized.
 class MemoryviewStream(io.IOBase):
-    # pyre-fixme[24]: Generic type `memoryview` expects 1 type parameter.
     def __init__(self, mv: memoryview) -> None:
-        # pyre-fixme[24]: Generic type `memoryview` expects 1 type parameter.
         self._mv: memoryview = mv.cast("b")
         self._pos = 0
 
-    # pyre-fixme[24]: Generic type `memoryview` expects 1 type parameter.
     def read(self, size: Optional[int] = -1) -> memoryview:
         if self.closed:
             raise ValueError("read from closed file")
@@ -41,7 +38,6 @@ def read(self, size: Optional[int] = -1) -> memoryview:
         self._pos = newpos
         return b
 
-    # pyre-fixme[24]: Generic type `memoryview` expects 1 type parameter.
     def read1(self, size: int = -1) -> memoryview:
         """This is the same as read."""
         return self.read(size)

torchsnapshot/serialization.py

@@ -173,7 +173,6 @@ class Serializer(Enum):
 ]
 
 
-# pyre-fixme[24]: Generic type `memoryview` expects 1 type parameter.
 def tensor_as_memoryview(tensor: torch.Tensor) -> memoryview:
     """
     Obtain the class::`memoryview` of a class::`torch.Tensor`.
@@ -200,11 +199,9 @@ def tensor_as_memoryview(tensor: torch.Tensor) -> memoryview:
         tensor = tensor.contiguous()
     if tensor.dtype == torch.bfloat16:
         return _tensor_as_memoryview_via_untyped_storage(tensor)
-    # pyre-fixme[6]: For 1st argument expected `Buffer` but got `ndarray[Any, Any]`.
     return memoryview(tensor.numpy()).cast("b")
 
 
-# pyre-fixme[24]: Generic type `memoryview` expects 1 type parameter.
 def _tensor_as_memoryview_via_untyped_storage(tensor: torch.Tensor) -> memoryview:
     """
     Obtain the class::`memoryview` of a class::`torch.Tensor` via untyped storage.
@@ -226,7 +223,6 @@ def _tensor_as_memoryview_via_untyped_storage(tensor: torch.Tensor) -> memoryvie
     untyped_storage = contiguous_view_as_untyped_storage(tensor)
     tensor = torch.empty((0))
     tensor.set_(untyped_storage)
-    # pyre-fixme[6]: For 1st argument expected `Buffer` but got `ndarray[Any, Any]`.
     return memoryview(tensor.numpy()).cast("b")
 
 
@@ -253,7 +249,6 @@ def contiguous_view_as_untyped_storage(tensor: torch.Tensor) -> UntypedStorage:
 
 
 def tensor_from_memoryview(
-    # pyre-fixme[24]: Generic type `memoryview` expects 1 type parameter.
     mv: memoryview,
     dtype: torch.dtype,
     shape: List[int],