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