import .pyi 文件无法导入

在服务器中运行别人的代码时用到了.pyi文件，但是显示no module

python环境已安装：Python 3.10.12

服务器版本：Ubuntu 22.04.3 LTS \n \l

在服务器实际运行时

>>> from mlir._mlir_libs._mlir import ir
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/workspace/tpu-mlir_v1.1.7-g30df2e3c-20230605/python/mlir/_mlir_libs/__init__.py", line 128, in <module>
    _site_initialize()
  File "/workspace/tpu-mlir_v1.1.7-g30df2e3c-20230605/python/mlir/_mlir_libs/__init__.py", line 56, in _site_initialize
    from ._mlir import ir
ImportError: cannot import name 'ir' from 'mlir._mlir_libs._mlir' (unknown location)

测试
>>> import ir
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'ir'

请问是需要安装什么编辑器吗？