runtime error

Exit code: 1. Reason: Traceback (most recent call last): File "/app/app.py", line 2, in <module> from utils import * File "/app/utils.py", line 3, in <module> from easyeditor import apply_grace_to_model, GraceHyperParams,nethook, apply_wise_to_model, WISEHyperParams, ROMEHyperParams, apply_rome_to_model File "/app/easyeditor/__init__.py", line 1, in <module> from .models import * File "/app/easyeditor/models/__init__.py", line 3, in <module> from .rome import * File "/app/easyeditor/models/rome/__init__.py", line 1, in <module> from .rome_main import ROMEHyperParams, apply_rome_to_model, execute_rome File "/app/easyeditor/models/rome/rome_main.py", line 10, in <module> from .compute_u import compute_u File "/app/easyeditor/models/rome/compute_u.py", line 11, in <module> from .layer_stats import layer_stats File "/app/easyeditor/models/rome/layer_stats.py", line 5, in <module> from datasets import load_dataset File "/usr/local/lib/python3.10/site-packages/datasets/__init__.py", line 33, in <module> from .arrow_dataset import Dataset, concatenate_datasets File "/usr/local/lib/python3.10/site-packages/datasets/arrow_dataset.py", line 60, in <module> from .arrow_writer import ArrowWriter, OptimizedTypedSequence File "/usr/local/lib/python3.10/site-packages/datasets/arrow_writer.py", line 25, in <module> from datasets.features.features import FeatureType, Value File "/usr/local/lib/python3.10/site-packages/datasets/features/__init__.py", line 3, in <module> from .features import * File "/usr/local/lib/python3.10/site-packages/datasets/features/features.py", line 478, in <module> class _ArrayXDExtensionType(pa.PyExtensionType): AttributeError: module 'pyarrow' has no attribute 'PyExtensionType'. Did you mean: 'ExtensionType'?

Container logs: