apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix tests

This commit is contained in:
Sean Sube 2023-12-21 09:51:06 -06:00
parent 4b430f4956
commit 855698cf39
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/tests/chain/test_blend_mask.py
View File

@ -4,7 +4,7 @@ from PIL import Image
from onnx_web.chain.blend_mask import BlendMaskStage
from onnx_web.chain.result import StageResult
from onnx_web.params import HighresParams, UpscaleParams
from onnx_web.params import HighresParams, SizeChart, UpscaleParams
class BlendMaskStageTests(unittest.TestCase):
@ -20,6 +20,7 @@ class BlendMaskStageTests(unittest.TestCase):
highres=HighresParams(False, 1, 0, 0),
upscale=UpscaleParams(""),
stage_mask=Image.new("RGBA", (64, 64)),
dims=(0, 0, SizeChart.auto)
)
self.assertEqual(len(result), 0)