Как импортировать AnalysisException в PySpark ⇐ Python

[Anonymous]

Я не могу найти, как импортировать AnalysisException в PySpark, чтобы его можно было перехватить. Например:
df = spark.createDataFrame([[1, 2], [1, 2]], ['A', 'A'])

try:
df.select('A')
except AnalysisException as e:
print(e)

Сообщение об ошибке:
NameError: name 'AnalysisException' is not defined


Подробнее здесь: https://stackoverflow.com/questions/654 ... in-pyspark