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