Skip to content

Commit 146f2f8

Browse files
authored
[ML] Bump client timeout for model download test (#116247) (#116584)
# Conflicts: # muted-tests.yml
1 parent 1192135 commit 146f2f8

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

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)