test pyre syntax

Differential Revision: D56491510
henrylhtsang · Apr 23, 2024 · f77a151 · f77a151
1 parent 0f3954f
commit f77a151
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/torchrec/distributed/tests/test_embedding_sharding.py b/torchrec/distributed/tests/test_embedding_sharding.py
@@ -36,6 +36,11 @@
 
 class TestGetWeightedAverageCacheLoadFactor(unittest.TestCase):
     def test_get_avg_cache_load_factor_hbm(self) -> None:
+        from typing import *
+
+        x: Optional[int | str] = None
+
+
         cache_load_factors = [random.random() for _ in range(5)]
         embedding_tables: List[ShardedEmbeddingTable] = [
             ShardedEmbeddingTable(