[java] fix double bracket init (#45460)

fixes https://errorprone.info/bugpattern/DoubleBraceInitialization Signed-off-by: Lonnie Liu <[email protected]>
ray-project · May 21, 2024 · 7767457 · 7767457
1 parent 592b5da
commit 7767457
Showing 1 changed file with 4 additions and 14 deletions.
diff --git a/java/test/src/main/java/io/ray/test/RuntimeEnvTest.java b/java/test/src/main/java/io/ray/test/RuntimeEnvTest.java
@@ -3,6 +3,7 @@
 import static io.ray.api.runtimeenv.types.RuntimeEnvName.JARS;
 
 import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableMap;
 import io.ray.api.ActorHandle;
 import io.ray.api.Ray;
 import io.ray.api.runtimeenv.RuntimeEnv;
@@ -72,18 +73,12 @@ public void testEnvVarsForNormalTask() {
       Ray.init();
       RuntimeEnv runtimeEnv = new RuntimeEnv.Builder().build();
       Map<String, String> envMap =
-          new HashMap<String, String>() {
-            {
-              put("KEY1", "A");
-              put("KEY2", "B");
-              put("KEY1", "C");
-            }
-          };
+          new HashMap<String, String>(ImmutableMap.of("KEY1", "A", "KEY2", "B", "KEY3", "C"));
       runtimeEnv.set(RuntimeEnvName.ENV_VARS, envMap);
 
       String val =
           Ray.task(RuntimeEnvTest::getEnvVar, "KEY1").setRuntimeEnv(runtimeEnv).remote().get();
-      Assert.assertEquals(val, "C");
+      Assert.assertEquals(val, "A");
       val = Ray.task(RuntimeEnvTest::getEnvVar, "KEY2").setRuntimeEnv(runtimeEnv).remote().get();
       Assert.assertEquals(val, "B");
     } finally {
@@ -97,12 +92,7 @@ public void testPerTaskEnvVarsOverwritePerJobEnvVars() {
     System.setProperty("ray.job.runtime-env.env-vars.KEY2", "B");
     try {
       Ray.init();
-      Map<String, String> envMap =
-          new HashMap<String, String>() {
-            {
-              put("KEY1", "C");
-            }
-          };
+      Map<String, String> envMap = new HashMap<String, String>(ImmutableMap.of("KEY1", "C"));
       RuntimeEnv runtimeEnv = new RuntimeEnv.Builder().build();
       runtimeEnv.set(RuntimeEnvName.ENV_VARS, envMap);