Fix @dataclass decorator bug reported at https://github.com/huggingface/transformers/issues/30412

modeling_cxrbert.py

@@ -14,11 +14,14 @@ from transformers.modeling_outputs import ModelOutput
 
 from .configuration_cxrbert import CXRBertConfig
 
+from dataclasses import dataclass # manually added due to this bug: https://github.com/huggingface/transformers/issues/30412
+
 BERTTupleOutput = Tuple[T, T, T, T, T]
 
+@dataclass # manually added due to this bug: https://github.com/huggingface/transformers/issues/30412
 class CXRBertOutput(ModelOutput):
-    last_hidden_state: torch.FloatTensor
-    logits: torch.FloatTensor
+    last_hidden_state: torch.FloatTensor = None # None added. Not present in the original code
+    logits: torch.FloatTensor = None # None added. Not present in the original code
     cls_projected_embedding: Optional[torch.FloatTensor] = None
     hidden_states: Optional[Tuple[torch.FloatTensor]] = None
     attentions: Optional[Tuple[torch.FloatTensor]] = None