Skip to content

Commit b205c02

Browse files
authored
[ML] Bump client timeout for model download test (#116247)
1 parent 97160ee commit b205c02

File tree

2 files changed

+10
-3
lines changed

2 files changed

+10
-3
lines changed

muted-tests.yml

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -207,9 +207,6 @@ tests:
207207
- class: org.elasticsearch.xpack.test.rest.XPackRestIT
208208
method: test {p0=ml/inference_crud/Test delete given model with alias referenced by pipeline}
209209
issue: https://github.com/elastic/elasticsearch/issues/116133
210-
- class: org.elasticsearch.xpack.inference.InferenceRestIT
211-
method: test {p0=inference/30_semantic_text_inference/Calculates embeddings using the default ELSER 2 endpoint}
212-
issue: https://github.com/elastic/elasticsearch/issues/114412
213210
- class: org.elasticsearch.xpack.esql.qa.multi_node.EsqlSpecIT
214211
method: test {categorize.Categorize SYNC}
215212
issue: https://github.com/elastic/elasticsearch/issues/113054

x-pack/plugin/inference/src/yamlRestTest/java/org/elasticsearch/xpack/inference/InferenceRestIT.java

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99

1010
import com.carrotsearch.randomizedtesting.annotations.ParametersFactory;
1111

12+
import org.elasticsearch.common.settings.Settings;
1213
import org.elasticsearch.test.cluster.ElasticsearchCluster;
1314
import org.elasticsearch.test.cluster.local.distribution.DistributionType;
1415
import org.elasticsearch.test.rest.yaml.ClientYamlTestCandidate;
@@ -30,6 +31,15 @@ public InferenceRestIT(final ClientYamlTestCandidate testCandidate) {
3031
super(testCandidate);
3132
}
3233

34+
@Override
35+
protected Settings restClientSettings() {
36+
var baseSettings = super.restClientSettings();
37+
return Settings.builder()
38+
.put(baseSettings)
39+
.put(CLIENT_SOCKET_TIMEOUT, "120s") // Long timeout for model download
40+
.build();
41+
}
42+
3343
@Override
3444
protected String getTestRestCluster() {
3545
return cluster.getHttpAddresses();

0 commit comments

Comments
 (0)