diff --git a/GANDLF/utils/write_parse.py b/GANDLF/utils/write_parse.py index b9d8fd3b4..c20654e7e 100644 --- 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"