Fixed commit d71ae5e

jpmml · Dec 25, 2023 · d9f5577 · d9f5577
1 parent d71ae5e
commit d9f5577
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pmml-sparkml/src/main/java/org/jpmml/sparkml/PipelineModelUtil.java b/pmml-sparkml/src/main/java/org/jpmml/sparkml/PipelineModelUtil.java
@@ -32,6 +32,7 @@
 import java.nio.file.Paths;
 import java.nio.file.SimpleFileVisitor;
 import java.nio.file.attribute.BasicFileAttributes;
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Enumeration;
 import java.util.List;
@@ -55,7 +56,7 @@ private PipelineModelUtil(){
 
 	static
 	public void addStage(PipelineModel pipelineModel, int index, Transformer transformer){
-		List<Transformer> stages = Arrays.asList(pipelineModel.stages());
+		List<Transformer> stages = new ArrayList<>(Arrays.asList(pipelineModel.stages()));
 
 		stages.add(index, transformer);
 
@@ -64,7 +65,7 @@ public void addStage(PipelineModel pipelineModel, int index, Transformer transfo
 
 	static
 	public Transformer removeStage(PipelineModel pipelineModel, int index){
-		List<Transformer> stages = Arrays.asList(pipelineModel.stages());
+		List<Transformer> stages = new ArrayList<>(Arrays.asList(pipelineModel.stages()));
 
 		Transformer result = stages.remove(index);