From eba10ac9838112dd8b29ce8f38218d9f2adbcd64 Mon Sep 17 00:00:00 2001
From: Andrew Beveridge <andrew@beveridge.uk>
Date: Thu, 14 Mar 2024 21:07:10 -0500
Subject: [PATCH] Fixed tests

---
 audio_separator/utils/cli.py | 8 +++-----
 tests/unit/test_cli.py       | 1 +
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/audio_separator/utils/cli.py b/audio_separator/utils/cli.py
index 23afd17..e665946 100755
--- a/audio_separator/utils/cli.py
+++ b/audio_separator/utils/cli.py
@@ -5,8 +5,6 @@
 import sys
 from importlib import metadata
 
-from audio_separator.separator import Separator
-
 
 def main():
     """Main entry point for the CLI."""
@@ -103,9 +101,7 @@ def main():
     mdxc_use_model_segment_size_help = "use model default segment size instead of the value from the config file. Example: --mdxc_use_model_segment_size"
     mdxc_overlap_help = "amount of overlap between prediction windows, 2-50. higher is better but slower (default: %(default)s). Example: --mdxc_overlap=8"
     mdxc_batch_size_help = "larger consumes more RAM but may process slightly faster (default: %(default)s). Example: --mdxc_batch_size=4"
-    mdxc_pitch_shift_help = (
-        "shift audio pitch by a number of semitones while processing. may improve output for deep/high vocals. (default: %(default)s). Example: --mdxc_pitch_shift=2"
-    )
+    mdxc_pitch_shift_help = "shift audio pitch by a number of semitones while processing. may improve output for deep/high vocals. (default: %(default)s). Example: --mdxc_pitch_shift=2"
 
     mdxc_params = parser.add_argument_group("MDXC Architecture Parameters")
     mdxc_params.add_argument("--mdxc_segment_size", type=int, default=256, help=mdxc_segment_size_help)
@@ -123,6 +119,8 @@ def main():
 
     logger.setLevel(log_level)
 
+    from audio_separator.separator import Separator
+
     if args.env_info:
         separator = Separator()
         sys.exit(0)
diff --git a/tests/unit/test_cli.py b/tests/unit/test_cli.py
index 216e209..658d14a 100644
--- a/tests/unit/test_cli.py
+++ b/tests/unit/test_cli.py
@@ -22,6 +22,7 @@ def common_expected_args():
         "mdx_params": {"hop_length": 1024, "segment_size": 256, "overlap": 0.25, "batch_size": 1, "enable_denoise": False},
         "vr_params": {"batch_size": 4, "window_size": 512, "aggression": 5, "enable_tta": False, "enable_post_process": False, "post_process_threshold": 0.2, "high_end_process": False},
         "demucs_params": {"segment_size": "Default", "shifts": 2, "overlap": 0.25, "segments_enabled": True},
+        "mdxc_params": {"segment_size": 256, "batch_size": 1, "overlap": 8, "use_model_segment_size": False, "pitch_shift": 0},
     }