diff --git a/.pyre_configuration b/.pyre_configuration index 1123fa52e..a453a78c7 100644 --- 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 index a242ad1fe..b1c9421d8 100644 --- 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 index 1b176fa4a..07c4d407c 100644 --- 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