aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--gn3/computations/qtlreaper.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/gn3/computations/qtlreaper.py b/gn3/computations/qtlreaper.py
index 49d363b..a88659e 100644
--- a/gn3/computations/qtlreaper.py
+++ b/gn3/computations/qtlreaper.py
@@ -27,7 +27,9 @@ def generate_traits_file(strains, trait_values, traits_filename):
header = "Traits\t{}\n".format("\t".join(strains))
data = [header] + [
"T{}\t{}\n".format(i+1, "\t".join([str(i) for i in t]))
- for i, t in enumerate(trait_values)]
+ for i, t in enumerate(trait_values[:-1])] + [
+ "T{}\t{}".format(len(trait_values), "\t".join([str(i) for i in t]))
+ for t in trait_values[-1:]]
with open(traits_filename, "w") as outfile:
outfile.writelines(data)