tuple compared to tuple

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
IBM · Jan 7, 2025 · bfbb78f · bfbb78f
1 parent d6486f3
commit bfbb78f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_encoder_decoder_model_factory.py b/tests/test_encoder_decoder_model_factory.py
@@ -181,7 +181,7 @@ def test_create_model_with_smp_deeplabv3plus_decoder(
     model.eval()
 
     with torch.no_grad():
-        assert model(model_input).output.shape == expected
+        assert model(model_input).output.shape == tuple(expected)
 
     gc.collect()