У меня была такая же ошибка, как и при использовании MLlib, и оказалось, что я вернул неправильный тип данных в одной из моих функций. Теперь он работает после простого нажатия на возвращаемое значение. Возможно, это не тот ответ, который вы ищете, но это, по крайней мере, намек на направление.