Remove Required hint from EncoderOutputDict

ludwig-ai · Jun 29, 2023 · 745375a · 745375a
1 parent e1a0dab
commit 745375a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ludwig/encoders/types.py b/ludwig/encoders/types.py
@@ -1,9 +1,9 @@
-from typing import Required, TypedDict
+from typing import TypedDict
 
 import torch
 
 
 class EncoderOutputDict(TypedDict, total=False):
-    encoder_output: Required[torch.Tensor]
+    encoder_output: torch.Tensor
     encoder_output_state: torch.Tensor
     attentions: torch.Tensor