Troubleshooting java.lang.IllegalArgumentException Error When Running Zingg Open Source on Databricks with Two Field ...

Hello, i'm trying to run Zingg open source using Databricks (by kinda following the example notebook) Im basically only matching two fields: field1: string that have varying size of 2 characters to 2000~3000 field2: int with sizes between 2 and 8 digits Im getting the error: java.lang.IllegalArgumentException: requirement failed: rawPredictionCol vectors must have length=2, but got 1 I have around 20k rows/records, did 66 matches and 33 non matches. Any tips?