diff --git a/finetune/tag_images_by_wd14_tagger.py b/finetune/tag_images_by_wd14_tagger.py
index 80a55716f..91e4f573e 100644
--- a/finetune/tag_images_by_wd14_tagger.py
+++ b/finetune/tag_images_by_wd14_tagger.py
@@ -224,7 +224,7 @@ def run_batch(path_imgs):
     print("done!")
 
 
-if __name__ == "__main__":
+def setup_parser() -> argparse.ArgumentParser:
     parser = argparse.ArgumentParser()
     parser.add_argument("train_data_dir", type=str, help="directory for train images / 学習画像データのディレクトリ")
     parser.add_argument(
@@ -284,6 +284,11 @@ def run_batch(path_imgs):
     )
     parser.add_argument("--frequency_tags", action="store_true", help="Show frequency of tags for images / 画像ごとのタグの出現頻度を表示する")
 
+    return parser
+
+if __name__ == "__main__":
+    parser = setup_parser()
+    
     args = parser.parse_args()
 
     # スペルミスしていたオプションを復元する