fix(__main__): replace pyinputplus with normal input (#587)

voicepaw · May 7, 2023 · 2b507da · 2b507da
1 parent e94ac6b
commit 2b507da
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/so_vits_svc_fork/__main__.py b/src/so_vits_svc_fork/__main__.py
@@ -7,7 +7,6 @@
 from typing import Literal
 
 import click
-import pyinputplus as pyip
 import torch
 
 from so_vits_svc_fork import __version__
@@ -784,7 +783,8 @@ def clean():
     import shutil
 
     folders = ["dataset", "filelists", "logs"]
-    if pyip.inputYesNo(f"Are you sure you want to delete files in {folders}?") == "yes":
+    # if pyip.inputYesNo(f"Are you sure you want to delete files in {folders}?") == "yes":
+    if input("Are you sure you want to delete files in {folders}?") in ["yes", "y"]:
         for folder in folders:
             if Path(folder).exists():
                 shutil.rmtree(folder)