Thanks for the amazing work! I am trying to build a prof of concept using huggingface models that might be replaced with custom build once later on.eleldar/theme-classification supports the arguments multi_class and candidate_labels but I am currently running into problems applying them correctly.

topics JSONB GENERATED ALWAYSAS (pgml.transform(    task=>'{       "task": "text-classification",               "model": "eleldar/theme-classification"        }'::JSONB,        args=>'{              "multi_class": true,              "candidate_labels": "[a, b,c]"        }'::JSONB,    inputs=> ARRAY[content])) STORED

This results in:

unexpected keyword argument 'multi_class'

What am i missing?