python-3.8 - impyla：如何设置 mem_limit？

Question

我在 python 3.8.3 上使用 impyla==0.16.2

试图执行set mem_limit=1G

并且在运行查询后它仍然给出 mem_limit 的错误。

这应该得到解决，因为如果我在 Dbeaver 上遵循相同的步骤，它会按预期工作。

不知道为什么它不能在 impyla 中工作。

有人可以指导我吗？

Answer 1

要使用 impala 设置 mem_limit，请将其作为确认参数传递给 cursor.execute 语句。

cursor = conn.cursor()
cursor.execute(query, configuration={'mem_limit': '1G'})

查询将成功运行。