Add ability to skip applications test via env var.

This commit is contained in:
Francois Chollet 2023-05-24 11:28:58 -07:00
parent e23c55d26d
commit 931e008683

@ -1,3 +1,5 @@
import os
import numpy as np
import pytest
from absl.testing import parameterized
@ -101,6 +103,10 @@ def _get_elephant(target_size):
return np.expand_dims(x, axis=0)
@pytest.mark.skipif(
os.environ.get("SKIP_APPLICATIONS_TESTS"),
reason="Env variable set to skip.",
)
class ApplicationsTest(testing.TestCase, parameterized.TestCase):
@parameterized.named_parameters(MODEL_LIST)
def test_application_notop_variable_input_channels(self, app, last_dim, _):