Skip to content

Commit 96915a0

Browse files
committed
style
1 parent 0c33bcc commit 96915a0

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/diffusers/utils/testing_utils.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -290,7 +290,10 @@ def decorator(test_case):
290290
return unittest.skip(test_case)
291291
else:
292292
current_compute_capability = get_torch_cuda_device_capability()
293-
return unittest.skipUnless(float(current_compute_capability) == float(expected_compute_capability), "Test not supported for this compute capability.")
293+
return unittest.skipUnless(
294+
float(current_compute_capability) == float(expected_compute_capability),
295+
"Test not supported for this compute capability.",
296+
)
294297

295298
return decorator
296299

0 commit comments

Comments
 (0)