Fix test + fix doc-string (#20)

* docstring change * update test
descriptinc · Jul 5, 2023 · 0202b4a · 0202b4a
1 parent 408235a
commit 0202b4a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/dac/utils/decode.py b/dac/utils/decode.py
@@ -122,7 +122,7 @@ def decode(
     device : str, optional
         Device to use, by default "cuda". If "cpu", the model will be loaded on the CPU.
     model_type : str, optional
-        The type of model to download. Must be one of "44khz", "24khz", or "16khz". Defaults to "44khz". Ignored if `weights_path` is specified.
+        The type of model to use. Must be one of "44khz", "24khz", or "16khz". Defaults to "44khz". Ignored if `weights_path` is specified.
     """
     generator = load_model(
         tag=model_tag,

diff --git a/dac/utils/encode.py b/dac/utils/encode.py
@@ -146,7 +146,7 @@ def encode(
     device : str, optional
         Device to use, by default "cuda"
     model_type : str, optional
-        The type of model to download. Must be one of "44khz", "24khz", or "16khz". Defaults to "44khz". Ignored if `weights_path` is specified.
+        The type of model to use. Must be one of "44khz", "24khz", or "16khz". Defaults to "44khz". Ignored if `weights_path` is specified.
     """
     generator = load_model(
         tag=model_tag,

diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -50,6 +50,7 @@ def test_reconstruction(model_type):
     args = {
         "input": str(input_dir),
         "output": str(output_dir),
+        "model_type": model_type,
     }
     with argbind.scope(args):
         run("decode")