From 3a026ea6e99ac713b3f76e8a41ff83caf63276eb Mon Sep 17 00:00:00 2001 From: "Pyre Bot Jr." Date: Thu, 17 Jul 2025 12:55:49 -0700 Subject: [PATCH] Add missing Pyre mode headers] [batch:1/N] [A] [A] (#3204) Summary: Pull Request resolved: https://github.com/pytorch/torchrec/pull/3204 X-link: https://github.com/pytorch/torchx/pull/1091 X-link: https://github.com/pytorch/tnt/pull/1016 Differential Revision: D78138183 --- torchrec/distributed/logger.py | 2 ++ torchrec/distributed/torchrec_logger.py | 2 ++ torchrec/modules/hash_mc_evictions.py | 2 ++ torchrec/modules/pruning_logger.py | 2 ++ 4 files changed, 8 insertions(+) diff --git a/torchrec/distributed/logger.py b/torchrec/distributed/logger.py index 5c6f6b7dd..bf42ded12 100644 --- a/torchrec/distributed/logger.py +++ b/torchrec/distributed/logger.py @@ -5,6 +5,8 @@ # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. +# pyre-strict + # mypy: allow-untyped-defs import functools import inspect diff --git a/torchrec/distributed/torchrec_logger.py b/torchrec/distributed/torchrec_logger.py index 90466be77..8cd49c34a 100644 --- a/torchrec/distributed/torchrec_logger.py +++ b/torchrec/distributed/torchrec_logger.py @@ -5,6 +5,8 @@ # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. +# pyre-strict + # mypy: allow-untyped-defs diff --git a/torchrec/modules/hash_mc_evictions.py b/torchrec/modules/hash_mc_evictions.py index 72ac84cd9..c5875bf8d 100644 --- a/torchrec/modules/hash_mc_evictions.py +++ b/torchrec/modules/hash_mc_evictions.py @@ -5,6 +5,8 @@ # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. +# pyre-strict + import logging import time from dataclasses import dataclass diff --git a/torchrec/modules/pruning_logger.py b/torchrec/modules/pruning_logger.py index 53bf1589e..09306cc2b 100644 --- a/torchrec/modules/pruning_logger.py +++ b/torchrec/modules/pruning_logger.py @@ -5,6 +5,8 @@ # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. +# pyre-strict + import logging from abc import ABC, abstractmethod from contextlib import contextmanager