Fix tests

This commit is contained in:
Francois Chollet 2023-05-18 18:03:34 -07:00
parent 44caf66041
commit ed4d749199
2 changed files with 2 additions and 1 deletions

@ -372,6 +372,7 @@ PYTHON_DTYPES_MAP = {
bool: "bool",
int: "int", # TBD by backend
float: "float32",
str: "string",
}

@ -65,7 +65,7 @@ class ZeroPadding3DTest(testing.TestCase, parameterized.TestCase):
self.assertAllClose(outputs[:, 2:-2, 2:-2, 2:-2, :], inputs)
@pytest.mark.skipif(
not backend.DYNAMIC_BATCH_SIZE_OK,
not backend.DYNAMIC_SHAPES_OK,
reason="Backend does not support dynamic batch sizes",
)
def test_zero_padding_3d_with_dynamic_spatial_dim(self):