Skip to content

Commit f61316a

Browse files
committed
Fix unexpected mode string print
- `AniCalcMode.fastani` => `fastani` - `AniCalcMode.skani` => `skani`
1 parent 74c4047 commit f61316a

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

src/aniclustermap/scripts/aniclustermap.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)