When transforming an ML Pipeline which uses StringIndexer, OneHotEncoder, VectorAssembler and LightGBMClassifier, it broke with "java.util.NoSuchElementException: Param