Py4JError: An error occurred while calling o511.array. Trace: