fine tune code

Signed-off-by: Yaliang Wu <[email protected]>
opensearch-project · Jul 12, 2023 · d6fb41c · d6fb41c
1 parent 5d7a898
commit d6fb41c
Showing 1 changed file with 1 addition and 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
@@ -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);