Fix OSS pyre

Summary: OSS CI Pyre lint failing Differential Revision: D57350370
sarckk · May 14, 2024 · 439a546 · 439a546
1 parent 7b8ace3
commit 439a546
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 6 deletions.
diff --git a/torchrec/distributed/benchmark/benchmark_utils.py b/torchrec/distributed/benchmark/benchmark_utils.py
@@ -576,9 +576,7 @@ def trace_handler(prof) -> None:
 
         with torch.profiler.profile(
             activities=[
-                # pyre-fixme[16]: Module `profiler` has no attribute `ProfilerActivity`.
                 torch.profiler.ProfilerActivity.CPU,
-                # pyre-fixme[16]: Module `profiler` has no attribute `ProfilerActivity`.
                 torch.profiler.ProfilerActivity.CUDA,
             ],
             record_shapes=True,

diff --git a/torchrec/distributed/sharding/rw_sequence_sharding.py b/torchrec/distributed/sharding/rw_sequence_sharding.py
@@ -299,8 +299,6 @@ def create_input_dist(
                 shard_split_offsets.append(table.global_metadata.size[0])
                 emb_sharding.extend([shard_split_offsets] * len(table.embedding_names))
 
-        # pyre-fixme[7]: Expected `BaseSparseFeaturesDist[KJTList]` but got
-        #  `InferCPURwSparseFeaturesDist`.
         return InferCPURwSparseFeaturesDist(
             world_size=self._world_size,
             num_features=num_features,

diff --git a/torchrec/distributed/types.py b/torchrec/distributed/types.py
@@ -859,8 +859,6 @@ def __init__(
         torch._C._log_api_usage_once(f"torchrec.distributed.{self.__class__.__name__}")
         self._qcomm_codecs_registry = qcomm_codecs_registry
 
-    # pyre-fixme[56]: Pyre doesn't yet support decorators with ParamSpec applied to
-    #  generic functions Please add # pyre-ignore[56] to `abc.abstractclassmethod`.
     @abc.abstractclassmethod
     # pyre-ignore [3]
     def shard(