runtime error

Exit code: 1. Reason: .arrow_dataset import Dataset File "/usr/local/lib/python3.10/site-packages/datasets/arrow_dataset.py", line 66, in <module> from .arrow_writer import ArrowWriter, OptimizedTypedSequence File "/usr/local/lib/python3.10/site-packages/datasets/arrow_writer.py", line 27, in <module> from .features import Features, Image, Value File "/usr/local/lib/python3.10/site-packages/datasets/features/__init__.py", line 18, in <module> from .features import Array2D, Array3D, Array4D, Array5D, ClassLabel, Features, Sequence, Value File "/usr/local/lib/python3.10/site-packages/datasets/features/features.py", line 629, in <module> class _ArrayXDExtensionType(pa.PyExtensionType): AttributeError: module 'pyarrow' has no attribute 'PyExtensionType'. Did you mean: 'ExtensionType'? The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/app/app.py", line 35, in <module> from egogpt.model.builder import load_pretrained_model File "/app/egogpt/model/__init__.py", line 1, in <module> from .language_model.egogpt_llama import EgoGPTConfig, EgoGPTLlamaForCausalLM File "/app/egogpt/model/language_model/egogpt_llama.py", line 30, in <module> from ..egogpt_arch import EgoGPTMetaForCausalLM, EgoGPTMetaModel File "/app/egogpt/model/egogpt_arch.py", line 25, in <module> from egogpt.utils import lengths_to_padding_mask, rank0_print, rank_print File "/app/egogpt/utils.py", line 272, in <module> def safe_save_model_for_hf_trainer(trainer: transformers.Trainer, output_dir: str): File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1525, in __getattr__ module = self._get_module(self._class_to_module[name]) File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1537, in _get_module raise RuntimeError( RuntimeError: Failed to import transformers.trainer because of the following error (look up to see its traceback): module 'pyarrow' has no attribute 'PyExtensionType'

Container logs: