From d6fb41c8297fc400225bb0b910d21488072a41de Mon Sep 17 00:00:00 2001 From: Yaliang Wu Date: Tue, 11 Jul 2023 21:05:43 -0700 Subject: [PATCH] fine tune code Signed-off-by: Yaliang Wu --- .../org/opensearch/ml/task/MLTrainingTaskRunnerTests.java | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/plugin/src/test/java/org/opensearch/ml/task/MLTrainingTaskRunnerTests.java b/plugin/src/test/java/org/opensearch/ml/task/MLTrainingTaskRunnerTests.java index c6a6b9a886..6565a41b95 100644 --- a/plugin/src/test/java/org/opensearch/ml/task/MLTrainingTaskRunnerTests.java +++ b/plugin/src/test/java/org/opensearch/ml/task/MLTrainingTaskRunnerTests.java @@ -112,11 +112,7 @@ public class MLTrainingTaskRunnerTests extends OpenSearchTestCase { public void setup() { MockitoAnnotations.openMocks(this); encryptor = new EncryptorImpl("0000000000000001"); - mlEngine = new MLEngine( - Path.of("/tmp/djl-cache_" + randomAlphaOfLength(10)), - Path.of("/tmp/djl-cache_" + randomAlphaOfLength(10)), - encryptor - ); + mlEngine = new MLEngine(Path.of("/tmp/djl-cache_" + randomAlphaOfLength(10)), encryptor); localNode = new DiscoveryNode("localNodeId", buildNewFakeTransportAddress(), Version.CURRENT); remoteNode = new DiscoveryNode("remoteNodeId", buildNewFakeTransportAddress(), Version.CURRENT); when(clusterService.localNode()).thenReturn(localNode);