-
Notifications
You must be signed in to change notification settings - Fork 8.3k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[ML] Functional tests - support tiny trained models #136010
Conversation
Pinging @elastic/ml-ui (:ml) |
@elasticmachine merge upstream |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Great job adding this in! LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Added a small comment but generally LGTM
Latest changes LGTM |
💚 Build SucceededMetrics [docs]
History
To update your PR or re-run it, just comment with: cc @pheyos |
Summary
This PR adds support for importing tiny trained PyTorch models during functional tests.
Details
x-pack/test/functional/services/ml/resources/trained_model_definitions
:pt_tiny_fill_mask
pt_tiny_ner
pt_tiny_pass_through
pt_tiny_text_classification
pt_tiny_text_embedding
pt_tiny_zero_shot
model_list
suite: import all 6 supported tiny models and checks row contents for all of themImported models during the test run look like this:
