fix(Assesment): Remove genes from stdout output

EUCANCan · Nov 16, 2023 · f9ea399 · f9ea399
1 parent a2bc9f9
commit f9ea399
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modules/oncoliner_assesment/src/assesment_main.py b/modules/oncoliner_assesment/src/assesment_main.py
@@ -158,7 +158,8 @@ def main(truth_vcf_paths, test_vcf_paths, output_prefix, fasta_ref, indel_thresh
     print(f'Total truth variants analyzed: {len(df_truth)} + {len(df_skipped_truth)} skipped')
     print(f'Total test variants analyzed: {len(df_test)} + {len(df_skipped_test)} skipped')
     print('Benchmark metrics:')
-    print(metrics_df.drop(['protein_affected_genes', 'protein_affected_driver_genes'], axis=1).to_string(index=False))
+    # Drop all columns that end with _genes
+    print(metrics_df.drop([col for col in metrics_df.columns if col.endswith('_genes')], axis=1).to_string(index=False))
     print(f'Benchmark metrics can be found in {output_prefix}metrics.csv')