@@ -154,22 +154,24 @@ def cli(
154154 if debug :
155155 tmpdir = outdir / "tmp"
156156 tmpdir .mkdir (exist_ok = True )
157- ani_matrix_tsv_file = outdir / f"{ mode } _matrix.tsv"
157+ ani_matrix_tsv_file = outdir / f"{ mode . value } _matrix.tsv"
158158 if not ani_matrix_tsv_file .exists () or overwrite :
159- logger .info (f"Run { mode } between all-vs-all genomes" )
159+ logger .info (f"Run { mode . value } between all-vs-all genomes" )
160160 AniCalcTool = dict (fastani = FastAni , skani = SkAni )[mode .value ]
161161 ani_matrix_df = AniCalcTool (indir , tmpdir ).run (thread_num = thread_num )
162162 ani_matrix_df .to_csv (ani_matrix_tsv_file , sep = "\t " , index = False )
163163 logger .info ("Write all-vs-all genomes ANI matrix result" )
164164 logger .info (f"=> { ani_matrix_tsv_file } " )
165165 else :
166- logger .info (f"Previous { mode } matrix result found (={ ani_matrix_tsv_file } )" )
167- logger .info (f"Skip { mode } run" )
166+ logger .info (
167+ f"Previous { mode .value } matrix result found (={ ani_matrix_tsv_file } )"
168+ )
169+ logger .info (f"Skip { mode .value } run" )
168170 ani_matrix_df = pd .read_csv (ani_matrix_tsv_file , sep = "\t " , index_col = False )
169171 ani_matrix_df = ani_matrix_df .set_index (ani_matrix_df .columns )
170172
171173 # Hierarchical clustering ANI matrix
172- logger .info (f"Clustering { mode } ANI matrix by scipy UPGMA method" )
174+ logger .info (f"Clustering { mode . value } ANI matrix by scipy UPGMA method" )
173175 linkage = hc .linkage (ani_matrix_df , method = "average" )
174176 dendrogram_newick_file = outdir / "ANIclustermap_dendrogram.nwk"
175177 to_newick_tree (ani_matrix_df , linkage , dendrogram_newick_file )
0 commit comments