Skip to content

Commit f250165

Browse files
robin-edesayakpaul
andauthored
Update tests/pipelines/qwenimage/test_qwenimage.py
Co-authored-by: Sayak Paul <[email protected]>
1 parent d77f5f5 commit f250165

File tree

1 file changed

+1
-11
lines changed

1 file changed

+1
-11
lines changed

tests/pipelines/qwenimage/test_qwenimage.py

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -260,14 +260,4 @@ def test_long_prompt_no_error(self):
260260
}
261261

262262
# This should not raise a RuntimeError about tensor dimension mismatch
263-
try:
264-
output = pipe(**inputs)
265-
# Basic sanity check that we got reasonable output
266-
self.assertIsNotNone(output)
267-
self.assertIsNotNone(output[0])
268-
except RuntimeError as e:
269-
if "must match the size of tensor" in str(e):
270-
self.fail(f"Long prompt caused dimension mismatch error: {e}")
271-
else:
272-
# Re-raise other runtime errors that aren't related to our fix
273-
raise
263+
_ = pipe(**inputs)

0 commit comments

Comments
 (0)