Pre-silence Pyre Errors for upcoming upgrade] [batch:58/603] [shard:3/N]

Reviewed By: MaggieMoss

Differential Revision: D65288809

fbshipit-source-id: 349ee2d6b0159744ae5ec1a4e317c6058d29f84e

Author: generatedunixname89002005287564

tests/test_memoryview_stream.py

@@ -17,6 +17,7 @@
 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,6 +180,7 @@ 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/io_types.py

@@ -17,6 +17,7 @@
 from .asyncio_utils import maybe_nested_loop
 
 
+# pyre-fixme[24]: Generic type `memoryview` expects 1 type parameter.
 BufferType = Union[bytes, memoryview]
 
 

torchsnapshot/memoryview_stream.py

@@ -13,10 +13,13 @@
 
 # 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")
@@ -38,6 +41,7 @@ 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,6 +173,7 @@ 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`.
@@ -199,9 +200,11 @@ 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.
@@ -223,6 +226,7 @@ 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")
 
 
@@ -249,6 +253,7 @@ 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],