Fix direct usage of TransformerModel

elastic / eland

Python Client and Toolkit for DataFrames, Big Data, Machine Learning and ETL in Elasticsearch

https://eland.readthedocs.io

Apache License 2.0

628 stars 98 forks source link

Fix direct usage of TransformerModel #619

Closed pquentin closed 9 months ago

pquentin commented 9 months ago

Closes #617

pquentin commented 9 months ago

Thanks. My intention is always to make a single change per pull request. In that case, there is indeed a single changed line in the core Eland package. The rest of the pull request fixes the tests that should have caught the bug but were skipped due to a pytest technicality. Does that make sense?

pquentin commented 9 months ago

Tested the latest commit with eland_import_hub_model and the PyTorchModel API, merging.