Skip to content

Commit b3a7825

Browse files
committed
Skip download check for non-packaged models
1 parent 602ae30 commit b3a7825

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/action/TransportPutTrainedModelAction.java

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -272,14 +272,18 @@ protected void masterOperation(
272272
}
273273
}, finalResponseListener::onFailure);
274274

275-
checkForExistingModelDownloadTask(
276-
client,
277-
trainedModelConfig.getModelId(),
278-
request.isWaitForCompletion(),
279-
finalResponseListener,
280-
() -> handlePackageAndTagsListener.onResponse(null),
281-
request.ackTimeout()
282-
);
275+
if (isPackageModel) {
276+
checkForExistingModelDownloadTask(
277+
client,
278+
trainedModelConfig.getModelId(),
279+
request.isWaitForCompletion(),
280+
finalResponseListener,
281+
() -> handlePackageAndTagsListener.onResponse(null),
282+
request.ackTimeout()
283+
);
284+
} else {
285+
handlePackageAndTagsListener.onResponse(null);
286+
}
283287
}
284288

285289
void verifyMlNodesAndModelArchitectures(

0 commit comments

Comments
 (0)