upgrade pyre version in fbcode/torchrec - batch 1 (pytorch#1557)

Summary: Pull Request resolved: pytorch#1557 Differential Revision: D51900798 fbshipit-source-id: 024498c892fca1dd0f8730b860d4e789d4c0ed7c
henrylhtsang · Dec 7, 2023 · a1b61b3 · a1b61b3
1 parent 1d20248
commit a1b61b3
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.pyre_configuration b/.pyre_configuration
@@ -13,5 +13,5 @@
     }
   ],
   "strict": true,
-  "version": "0.0.101698750730"
+  "version": "0.0.101701778454"
 }
diff --git a/torchrec/distributed/tests/test_dist_data.py b/torchrec/distributed/tests/test_dist_data.py
@@ -184,7 +184,6 @@ def _generate_variable_batch_sparse_features_batch(
         out_jagged.append(
             KeyedJaggedTensor.from_lengths_sync(
                 keys=out_keys,
-                # pyre-ignore[6]
                 stride_per_key_per_rank=[
                     list(_flatten(batch_sizes_by_rank[key_idx]))
                     for key_idx in key_indices

diff --git a/torchrec/distributed/tests/test_lazy_awaitable.py b/torchrec/distributed/tests/test_lazy_awaitable.py
@@ -242,6 +242,8 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
 
         tempFile = None
         with tempfile.NamedTemporaryFile(delete=False) as f:
+            # pyre-fixme[6]: For 2nd argument expected `SupportsWrite[bytes]` but
+            #  got `_TemporaryFileWrapper[bytes]`.
             pickle.dump(gm, f)
             tempFile = f