Hi,

I am doing a grammar check using spark NLP using azure databricks. But am getting TypeError: 'JavaPackage' object is not callable - in DocumentAssembler() intialization line.

document_assembler = DocumentAssembler()\

 .setInputCol("text")\

 .setOutputCol("document")

Same code I tried with colab notebook its working fine.

Is anybody please help me on this?

sparknlp version : 4.2.3

pyspark version : 3.1.2