github-mirrors/elasticsearch
1
0
Fork 0
mirror of https://github.com/elastic/elasticsearch.git synced 2025-06-28 09:28:55 -04:00
Code Issues Projects Releases Packages Wiki Activity

[ML] Avoid potentially throwing calls to Task#getDescription in model download

Browse source
This commit is contained in:
David Kyle 2025-03-11 09:48:07 +00:00 • committed by GitHub
parent c26d195120
commit 444b8eab75
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
docs/changelog/124527.yaml Normal file
View file

@ -0,0 +1,5 @@
pr: 124527
summary: Avoid potentially throwing calls to Task#getDescription in model download
area: Machine Learning
type: bug
issues: []

2
x-pack/plugin/ml-package-loader/src/main/java/org/elasticsearch/xpack/ml/packageloader/action/TransportLoadTrainedModelPackage.java
View file

@ -151,7 +151,7 @@ public class TransportLoadTrainedModelPackage extends TransportMasterNodeAction<
ModelDownloadTask inProgress = null;
for (var task : tasks) {
if (description.equals(task.getDescription()) && task instanceof ModelDownloadTask downloadTask) {
if (task instanceof ModelDownloadTask downloadTask && (description.equals(downloadTask.getDescription()))) {
inProgress = downloadTask;
break;
}