Pyre Configurationless migration for] [batch:28/29]

Reviewed By: connernilsen Differential Revision: D56626638 fbshipit-source-id: aa2621558977fba39bf29d43000a8de43d0d87c0
pytorch · Apr 26, 2024 · ae3de28 · ae3de28
1 parent ab7a9fe
commit ae3de28
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/torchrec/sparse/tests/test_jagged_tensor_gpu.py b/torchrec/sparse/tests/test_jagged_tensor_gpu.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 unittest
 

diff --git a/torchrec/sparse/tests/utils.py b/torchrec/sparse/tests/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
+
 import random
 from typing import List