Pyre Configurationless migration for] [batch:8/29] (#1896)

Summary: Pull Request resolved: #1896 Reviewed By: connernilsen Differential Revision: D56298191 fbshipit-source-id: 3d3f86e5394e21facd18e5939ee42aa7da18d33f
pytorch · Apr 18, 2024 · a80219a · a80219a
1 parent 59878e8
commit a80219a
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 0 deletions.
diff --git a/torchrec/distributed/infer_utils.py b/torchrec/distributed/infer_utils.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
+
 from typing import List, Optional, Set, Tuple
 
 import torch

diff --git a/torchrec/distributed/tests/test_infer_utils.py b/torchrec/distributed/tests/test_infer_utils.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
+
 #!/usr/bin/env python3
 
 import unittest

diff --git a/torchrec/ir/types.py b/torchrec/ir/types.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
+
 #!/usr/bin/env python3
 
 import abc

diff --git a/torchrec/ir/utils.py b/torchrec/ir/utils.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
+
 #!/usr/bin/env python3
 
 from typing import Type