diff --git a/torchrec/sparse/tests/test_jagged_tensor_gpu.py b/torchrec/sparse/tests/test_jagged_tensor_gpu.py
index 4e5975ae1..b32075f8d 100644
--- 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
index 46899b6c6..d8ae4cad5 100644
--- 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