diff --git a/ludwig/encoders/types.py b/ludwig/encoders/types.py
index 24e4a086c6d..946086f64b9 100644
--- 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