File tree Expand file tree Collapse file tree 1 file changed +3
-6
lines changed
Expand file tree Collapse file tree 1 file changed +3
-6
lines changed Original file line number Diff line number Diff line change @@ -299,15 +299,12 @@ def __init__(
299299 ):
300300 if all (isinstance (input , PIL .Image .Image ) for input in [actual , expected ]):
301301 actual , expected = (to_image (input ) for input in [actual , expected ])
302-
303- if CVCUDA_AVAILABLE and all (isinstance (input , _import_cvcuda ().Tensor ) for input in [actual , expected ]):
302+ elif CVCUDA_AVAILABLE and all (isinstance (input , _import_cvcuda ().Tensor ) for input in [actual , expected ]):
304303 actual , expected = (cvcuda_to_tensor (input ) for input in [actual , expected ])
305-
306- if CVCUDA_AVAILABLE and isinstance (actual , _import_cvcuda ().Tensor ) and isinstance (expected , PIL .Image .Image ):
304+ elif CVCUDA_AVAILABLE and isinstance (actual , _import_cvcuda ().Tensor ) and isinstance (expected , PIL .Image .Image ):
307305 actual = cvcuda_to_pil_compatible_tensor (actual )
308306 expected = to_image (expected )
309-
310- if CVCUDA_AVAILABLE and isinstance (actual , _import_cvcuda ().Tensor ):
307+ elif CVCUDA_AVAILABLE and isinstance (actual , _import_cvcuda ().Tensor ):
311308 actual = cvcuda_to_pil_compatible_tensor (actual )
312309
313310 super ().__init__ (actual , expected , ** other_parameters )
You can’t perform that action at this time.
0 commit comments