From 8af5df491efc0fea6e83dbe43a6050f080bfa18e Mon Sep 17 00:00:00 2001 From: Qing Date: Mon, 27 Mar 2023 20:54:05 +0800 Subject: [PATCH] fix test --- lama_cleaner/tests/test_model.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/lama_cleaner/tests/test_model.py b/lama_cleaner/tests/test_model.py index a76f3c6..be253b5 100644 --- a/lama_cleaner/tests/test_model.py +++ b/lama_cleaner/tests/test_model.py @@ -132,16 +132,20 @@ def test_zits(strategy, zits_wireframe): ) -@pytest.mark.parametrize("strategy", [HDStrategy.ORIGINAL], "no_half", [True, False]) +@pytest.mark.parametrize("strategy", [HDStrategy.ORIGINAL]) +@pytest.mark.parametrize("no_half", [True, False]) def test_mat(strategy, no_half): model = ModelManager(name="mat", device=device, no_half=no_half) cfg = get_config(strategy) for _ in range(10): assert_equal( - model, cfg, f"mat_{strategy.capitalize()}_result.png", + model, + cfg, + f"mat_{strategy.capitalize()}_result.png", ) + @pytest.mark.parametrize("strategy", [HDStrategy.ORIGINAL]) def test_fcf(strategy): model = ModelManager(name="fcf", device=device) @@ -188,4 +192,3 @@ def test_manga(strategy): img_p=current_dir / "overture-creations-5sI6fQgYIuo.png", mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png", ) ->>>>>>> 944494a (mat support float16)