Do for each iteration

Author: IoannisPanagiotas

algo/src/main/java/org/neo4j/gds/embeddings/node2vec/Node2VecModel.java

@@ -128,6 +128,8 @@ Node2VecResult train() {
 
         var lossPerIteration = new ArrayList<Double>();
 
+        AtomicInteger taskIndex = new AtomicInteger(0);
+
         for (int iteration = 0; iteration < iterations; iteration++) {
             progressTracker.beginSubTask();
             progressTracker.setVolume(walks.size());
@@ -137,7 +139,7 @@ Node2VecResult train() {
                 initialLearningRate - iteration * learningRateAlpha
             );
 
-            var tasks = createTrainingTasks(learningRate);
+            var tasks = createTrainingTasks(learningRate, taskIndex);
 
             RunWithConcurrency.builder()
                 .concurrency(concurrency)
@@ -288,8 +290,7 @@ void addAll(FloatConsumer other) {
         }
     }
 
-    List<TrainingTask> createTrainingTasks(float learningRate){
-        AtomicInteger taskIndex = new AtomicInteger(0);
+    List<TrainingTask> createTrainingTasks(float learningRate, AtomicInteger taskIndex){
         return PartitionUtils.degreePartitionWithBatchSize(
             walks.size(),
             walks::walkLength,