Text expansion not supported when using 3rd party models

[shahkrohan]

trafficstars

File "/env/lib/conda/rohan-retrieval/lib/python3.10/site-packages/eland/ml/pytorch/transformers.py", line 665, in init self._traceable_model = self._create_traceable_model() File "/env/lib/conda/rohan-retrieval/lib/python3.10/site-packages/eland/ml/pytorch/transformers.py", line 1029, in _create_traceable_model raise TypeError( TypeError: Unknown task type text_expansion, must be one of: fill_mask, ner, pass_through, question_answering, text_classification, text_embedding, text_expansion, text_similarity, zero_shot_classification

this is from here. would appreciate if you fixed this ASAP

[pquentin]

Thanks! It looks like @davidkyle may have missed that in #520? This is indeed the only task type missing in _create_traceable_model.

[rohankshah04]

#706

[davidkyle]

Closed by #740