提交 5fb2e8e1 编写于 作者: M Megvii Engine Team 提交者: Wanwan1996

fix(data): fix pyarrow.plasma import error in pyarrow1.12

GitOrigin-RevId: b5e1cd3be59cc80a3cc5bf6a83855ede2a2cd38a
上级 cc7b2f16
......@@ -27,7 +27,6 @@ try:
except:
import _thread as thread
logger = get_logger(__name__)
......
......@@ -9,6 +9,13 @@ import pyarrow
MGE_PLASMA_MEMORY = int(os.environ.get("MGE_PLASMA_MEMORY", 4000000000)) # 4GB
try:
import pyarrow.plasma as plasma
except ModuleNotFoundError:
raise RuntimeError(
"pyarrow remove plasma in version 12.0.0, please use pyarrow vserion < 12.0.0"
)
try:
import pyarrow.plasma as plasma
except ModuleNotFoundError:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册