found last error in test using xlrd

openvax · May 9, 2024 · 74fcfd5 · 74fcfd5
1 parent fd57d67
commit 74fcfd5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/test_shell_script.py b/tests/test_shell_script.py
@@ -17,6 +17,7 @@
 from tempfile import NamedTemporaryFile
 
 import pandas as pd
+# TODO: change this to use openpyxl
 from xlrd import open_workbook
 
 from vaxrank.cli import main as run_shell_script

diff --git a/vaxrank/report.py b/vaxrank/report.py
@@ -454,8 +454,9 @@ def resize_columns(worksheet, amino_acids_col, pos_col):
     Resizes amino acid and mutant position columns in the Excel sheet so that they don't
     have to be expanded.
     """
-    worksheet.set_column('%s:%s' % (amino_acids_col, amino_acids_col), 40)
-    worksheet.set_column('%s:%s' % (pos_col, pos_col), 12)
+    worksheet.column_dimensions[amino_acids_col].width = 40
+    worksheet.column_dimensions[pos_col].width = 12
+
 
 def make_minimal_neoepitope_report(
         ranked_variants_with_vaccine_peptides,