Skip to content

Commit

Permalink
[ML] Mute PyTorch tests (elastic#86263)
Browse files Browse the repository at this point in the history
davidkyle authored Apr 28, 2022

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
1 parent d742be8 commit 74d86b5
Showing 3 changed files with 5 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -57,6 +57,7 @@ public void cleanUpAfterTest() {
cleanUp();
}

@AwaitsFix(bugUrl = "https://github.com/elastic/ml-cpp/pull/2258")
public void testMemoryStats() throws Exception {

deployTrainedModel();
Original file line number Diff line number Diff line change
@@ -8,6 +8,7 @@
package org.elasticsearch.xpack.ml.integration;

import org.apache.http.util.EntityUtils;
import org.apache.lucene.tests.util.LuceneTestCase;
import org.elasticsearch.client.Request;
import org.elasticsearch.client.Response;
import org.elasticsearch.client.ResponseException;
@@ -74,6 +75,8 @@
* torch.jit.save(traced_model, "simplemodel.pt")
* ## End Python
*/

@LuceneTestCase.AwaitsFix(bugUrl = "https://github.com/elastic/ml-cpp/pull/2258")
public class PyTorchModelIT extends ESRestTestCase {

private static final String BASIC_AUTH_VALUE_SUPER_USER = UsernamePasswordToken.basicAuthHeaderValue(
Original file line number Diff line number Diff line change
@@ -165,6 +165,7 @@ public void testMLFeatureResetFailureDueToPipelines() throws Exception {
assertThat(isResetMode(), is(false));
}

@AwaitsFix(bugUrl = "https://github.com/elastic/ml-cpp/pull/2258")
public void testMLFeatureResetWithModelDeployment() throws Exception {
createModelDeployment();
client().execute(ResetFeatureStateAction.INSTANCE, new ResetFeatureStateRequest()).actionGet();

0 comments on commit 74d86b5

Please sign in to comment.