fix(controller): Collector.toMap require non-null value (#3020)

star-whale · Nov 23, 2023 · 1f2320e · 1f2320e
1 parent 5d8927d
commit 1f2320e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/server/controller/src/main/java/ai/starwhale/mlops/domain/ft/FineTuneAppService.java b/server/controller/src/main/java/ai/starwhale/mlops/domain/ft/FineTuneAppService.java
@@ -238,7 +238,9 @@ private String rewriteSpecEnvForRequest(
                 if (null == env) {
                     env = new ArrayList<>();
                 }
-                Map<String, String> envMap = env.stream().collect(Collectors.toMap(Env::getName, Env::getValue));
+                Map<String, String> envMap = env.stream()
+                        .filter(e -> null != e.getValue())
+                        .collect(Collectors.toMap(Env::getName, Env::getValue));
                 //if ENV vars in user's step spec conflicts with controller's, use controller's ENV
                 List<Env> controllerEnvs = envSupplier.get();
                 if (null != controllerEnvs) {