mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-10 06:10:50 +08:00
Fix tests (no rerelease necessary)
This commit is contained in:
parent
cb97b94ad9
commit
7f7bcd5d8f
@ -122,7 +122,8 @@ def test_binary_operation_lazy_check():
|
||||
|
||||
# Test standard ops
|
||||
assert n.check_lazy_status(op="eq") == ["lhs", "rhs"]
|
||||
assert n.check_lazy_status(op="add", lhs=1, rhs=2) == ["lhs", "rhs"]
|
||||
assert n.check_lazy_status(op="add", lhs=1, rhs=2) == []
|
||||
assert n.check_lazy_status(op="add", lhs=None, rhs=None) == ["lhs", "rhs"]
|
||||
|
||||
# Test 'and'
|
||||
assert n.check_lazy_status(op="and", lhs=None) == ["lhs"]
|
||||
|
||||
@ -91,7 +91,7 @@ def test_sdpa_import_exception():
|
||||
importlib.reload(comfy.ops)
|
||||
|
||||
assert comfy.ops.scaled_dot_product_attention is comfy.ops._scaled_dot_product_attention
|
||||
mock_logger.debug.assert_called_once()
|
||||
mock_logger.debug.assert_called()
|
||||
# Check that the log message contains the exception info
|
||||
assert "Could not set sdpa backend priority." in mock_logger.debug.call_args[0][0]
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user