From 0202b4a71a80e4d6b663a898b65c4b6b8b1f4c14 Mon Sep 17 00:00:00 2001 From: Ishaan Kumar Date: Wed, 5 Jul 2023 12:09:28 -0400 Subject: [PATCH] Fix test + fix doc-string (#20) * docstring change * update test --- dac/utils/decode.py | 2 +- dac/utils/encode.py | 2 +- tests/test_cli.py | 1 + 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/dac/utils/decode.py b/dac/utils/decode.py index c8f32ef..a3597e2 100644 --- 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 index 0f1f620..b33eb13 100644 --- 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 index a39cc26..9487b66 100644 --- 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")