test pyre syntax (pytorch#1922)

Summary: Pull Request resolved: pytorch#1922 Differential Revision: D56491510
henrylhtsang · Apr 24, 2024 · 21a772c · 21a772c
1 parent 76e854c
commit 21a772c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/torchrec/distributed/tests/test_embedding_sharding.py b/torchrec/distributed/tests/test_embedding_sharding.py
@@ -10,7 +10,7 @@
 
 import random
 import unittest
-from typing import Any, Dict, List
+from typing import Any, Dict, List, Optional
 from unittest.mock import MagicMock
 
 import hypothesis.strategies as st
@@ -36,6 +36,9 @@
 
 class TestGetWeightedAverageCacheLoadFactor(unittest.TestCase):
     def test_get_avg_cache_load_factor_hbm(self) -> None:
+        x: Optional[int | str] = None
+        _ = x or 1
+
         cache_load_factors = [random.random() for _ in range(5)]
         embedding_tables: List[ShardedEmbeddingTable] = [
             ShardedEmbeddingTable(