Fix test + fix doc-string (#20)

eeishaan · web-flow · commit 0202b4a71a80 · 2023-07-05T12:09:28.000-04:00
* docstring change

* update test
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")

Original file line number	Diff line number	Diff line change
`@@ -50,6 +50,7 @@ def test_reconstruction(model_type):`
`50`	`50`	`args = {`
`51`	`51`	`"input": str(input_dir),`
`52`	`52`	`"output": str(output_dir),`
	`53`	`+ "model_type": model_type,`
`53`	`54`	`}`
`54`	`55`	`with argbind.scope(args):`
`55`	`56`	`run("decode")`