runtime error

Exit code: 1. Reason: d_import File "<frozen importlib._bootstrap>", line 1027, in _find_and_load File "<frozen importlib._bootstrap>", line 1006, in _find_and_load_unlocked File "<frozen importlib._bootstrap>", line 688, in _load_unlocked File "<frozen importlib._bootstrap_external>", line 883, in exec_module File "<frozen importlib._bootstrap>", line 241, in _call_with_frames_removed File "/home/user/app/server/pipelines/controlnet.py", line 20, in <module> from pruna import SmashConfig, smash File "/usr/local/lib/python3.10/dist-packages/pruna/__init__.py", line 16, in <module> from pruna.config.smash_config import SmashConfig File "/usr/local/lib/python3.10/dist-packages/pruna/config/smash_config.py", line 34, in <module> from pruna.data.pruna_datamodule import PrunaDataModule, TokenizerMissingError File "/usr/local/lib/python3.10/dist-packages/pruna/data/__init__.py", line 17, in <module> from pruna.data.datasets.audio import setup_commonvoice_dataset, setup_podcast_dataset File "/usr/local/lib/python3.10/dist-packages/pruna/data/datasets/audio.py", line 19, in <module> from datasets import Dataset, load_dataset File "/usr/local/lib/python3.10/dist-packages/datasets/__init__.py", line 22, in <module> from .arrow_dataset import Dataset File "/usr/local/lib/python3.10/dist-packages/datasets/arrow_dataset.py", line 67, in <module> from .arrow_writer import ArrowWriter, OptimizedTypedSequence File "/usr/local/lib/python3.10/dist-packages/datasets/arrow_writer.py", line 27, in <module> from .features import Features, Image, Value File "/usr/local/lib/python3.10/dist-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/dist-packages/datasets/features/features.py", line 634, in <module> class _ArrayXDExtensionType(pa.PyExtensionType): AttributeError: module 'pyarrow' has no attribute 'PyExtensionType'. Did you mean: 'ExtensionType'?

Container logs: