df: Fix arguments for deepFilter script

Rikorose · May 6, 2022 · afe6bfb · afe6bfb
1 parent 5b28635
commit afe6bfb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/DeepFilterNet/df/enhance.py b/DeepFilterNet/df/enhance.py
@@ -274,7 +274,7 @@ def setup_df_argument_parser() -> argparse.ArgumentParser:
     return parser
 
 
-if __name__ == "__main__":
+def run():
     parser = setup_df_argument_parser()
     parser.add_argument(
         "--compensate-delay",
@@ -289,3 +289,7 @@ def setup_df_argument_parser() -> argparse.ArgumentParser:
         help="List of noise files to mix with the clean speech file.",
     )
     main(parser.parse_args())
+
+
+if __name__ == "__main__":
+    run()
diff --git a/DeepFilterNet/pyproject.toml b/DeepFilterNet/pyproject.toml
@@ -41,7 +41,7 @@ semetrics = ["semetrics"]
 soundfile = ["soundfile"]
 
 [tool.poetry.scripts]
-deepFilter = "df.enhance:main"
+deepFilter = "df.enhance:run"
 
 [tool.poetry.dev-dependencies]
 poethepoet = "^0.11"