[SOURCE] Fix RR batch ID issue

- TODO fix batch ID issue of random
leondavi · Jul 20, 2024 · 1dd4250 · 1dd4250
1 parent e46270b
commit 1dd4250
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src_erl/NerlnetApp/src/Source/sourceSendingPolicies.erl b/src_erl/NerlnetApp/src/Source/sourceSendingPolicies.erl
@@ -73,7 +73,7 @@ send_method_round_robin(TransmitterEts, Epochs, TimeInterval_ms, ClientWorkerPai
   TotalNumOfBatches = length(BatchesListToSend),
   BatchesIndexes = generate_batch_indexes(TotalNumOfBatches, EpochIdx),
   ClientWorkerPairsLength = length(ClientWorkerPairs),
-  WorkerIdxBatchIdxTuples = [ {X , X rem ClientWorkerPairsLength} || X <- BatchesIndexes],
+  WorkerIdxBatchIdxTuples = [ {X div ClientWorkerPairsLength, X rem ClientWorkerPairsLength} || X <- BatchesIndexes],
   BatchesWithIndexes = lists:zip(WorkerIdxBatchIdxTuples, BatchesListToSend),
   lists:foreach(BatchFunc, BatchesWithIndexes),
   % update batches sent
@@ -97,7 +97,7 @@ send_method_random(TransmitterEts, Epochs, TimeInterval_ms, ClientWorkerPairs, B
   TotalNumOfBatches = length(BatchesListToSend),
   BatchesIndexes = generate_batch_indexes(TotalNumOfBatches, EpochIdx),
   ClientWorkerPairsLength = length(ClientWorkerPairs),
-  WorkerIdxBatchIdxTuples = [ {X , rand:uniform(ClientWorkerPairsLength)} || X <- BatchesIndexes],
+  WorkerIdxBatchIdxTuples = [ {X div ClientWorkerPairsLength, rand:uniform(ClientWorkerPairsLength)} || X <- BatchesIndexes],
   BatchesWithIndexes = lists:zip(WorkerIdxBatchIdxTuples, BatchesListToSend), % Tuple {{BatchIdx, WorkerIdx}, Batch}
   lists:foreach(BatchFunc, BatchesWithIndexes),
   % update batches sent