fix py3.9 build

Summary: The `|` operator is only supported for python >= 3.10, which breaks python 3.9 build Differential Revision: D67723223
sarckk · Dec 30, 2024 · 2656ad4 · 2656ad4
1 parent f059a49
commit 2656ad4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torchrec/distributed/mc_modules.py b/torchrec/distributed/mc_modules.py
@@ -66,7 +66,9 @@
 
 @dataclass
 class EmbeddingCollectionContext(Multistreamable):
-    sharding_contexts: List[InferSequenceShardingContext | SequenceShardingContext]
+    sharding_contexts: List[
+        Union[InferSequenceShardingContext, SequenceShardingContext]
+    ]
 
     def record_stream(self, stream: torch.Stream) -> None:
         for ctx in self.sharding_contexts: