use join

mlcommons · Aug 23, 2023 · 03a8e9d · 03a8e9d
1 parent e42962a
commit 03a8e9d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/GANDLF/utils/write_parse.py b/GANDLF/utils/write_parse.py
@@ -22,8 +22,7 @@ def writeTrainingCSV(
     channelsID_list = channelsID.split(",")  # split into list
 
     outputToWrite = "SubjectID,"
-    for i, n in enumerate(channelsID_list):
-        outputToWrite += "Channel_" + str(i) + ","
+    outputToWrite += ",".join(["Channel_" + str(i) for i, n in enumerate(channelsID_list)])
     if labelID is not None:
         outputToWrite += "Label"
     outputToWrite += "\n"