diff --git a/vaxrank/report.py b/vaxrank/report.py index 9b4560f..8fe21aa 100644 --- a/vaxrank/report.py +++ b/vaxrank/report.py @@ -426,6 +426,7 @@ def new_columns(): ("pos", []), ("ref", []), ("alt", []), + ("gene_name", []), ("variant_rank", []), ("peptide_rank", []), ("mutation_start", []), @@ -512,7 +513,7 @@ def make_minimal_neoepitope_report( worksheet.set_column('%s:%s' % ('G', 'G'), 30) worksheet.set_column('%s:%s' % ('H', 'H'), 9) worksheet.set_column('%s:%s' % ('I', 'I'), 18) - writer.save() + writer.close() logger.info('Wrote XLSX neoepitope report file to %s', excel_report_path) @@ -546,6 +547,7 @@ def make_csv_report( columns["pos"].append(variant.original_start) columns["ref"].append(variant.original_ref) columns["alt"].append(variant.original_alt) + columns["gene_name"].append(vaccine_peptide.mutant_protein_fragment.gene_name) columns["variant_rank"].append(i + 1) columns["peptide_rank"].append(j + 1) columns["amino_acids"].append(vaccine_peptide.mutant_protein_fragment.amino_acids) @@ -596,5 +598,5 @@ def make_csv_report( df.to_excel(writer, sheet_name=shortened_sheet_name, index=False) resize_columns(writer.sheets[shortened_sheet_name], 'A', 'C') - writer.save() + writer.close() logger.info('Wrote manufacturer XLSX file to %s', excel_report_path)