Add missing Pyre mode headers] [batch:1/N] [A] [A]

Summary:
X-link: meta-pytorch/torchx#1091

X-link: meta-pytorch/tnt#1016

Differential Revision: D78138183

Author: Pyre Bot Jr.

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

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
 
 

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

torchrec/modules/hash_mc_modules.py

@@ -6,6 +6,7 @@
 # LICENSE file in the root directory of this source tree.
 
 # pyre-strict
+
 import logging
 import math
 from typing import Any, Dict, Iterator, List, Optional, Tuple

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