Skip to content

Commit

Permalink
[CI] Move mask_rcnn model to nightly TF tests (#2453)
Browse files Browse the repository at this point in the history
### Changes

- TF tests: Add nightly mark for mask_rcnn model.
- Reduced time of TF precommit from ~65min to ~30min.

### Reason for changes

Skip long-running tests in precommit
  • Loading branch information
ksilligan authored Mar 4, 2024
1 parent 2cb582f commit fae793d
Show file tree
Hide file tree
Showing 4 changed files with 10 additions and 3 deletions.
7 changes: 6 additions & 1 deletion Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -102,7 +102,12 @@ install-tensorflow-dev: install-tensorflow-test install-pre-commit
pip install -r examples/post_training_quantization/tensorflow/mobilenet_v2/requirements.txt

test-tensorflow:
pytest ${COVERAGE_ARGS} tests/tensorflow \
pytest ${COVERAGE_ARGS} tests/tensorflow -m "not nightly" \
--junitxml ${JUNITXML_PATH} \
$(DATA_ARG)

test-tensorflow-nightly:
pytest ${COVERAGE_ARGS} tests/tensorflow -m 'nightly' \
--junitxml ${JUNITXML_PATH} \
$(DATA_ARG)

Expand Down
1 change: 1 addition & 0 deletions tests/tensorflow/pytest.ini
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
[pytest]
markers =
install
nightly
4 changes: 3 additions & 1 deletion tests/tensorflow/test_compressed_graph.py
Original file line number Diff line number Diff line change
Expand Up @@ -199,11 +199,13 @@ def __init__(
"nasnet_mobile": pytest.mark.skip(reason="gitlab issue #18"),
"mobilenet_v2_slim": pytest.mark.skip(reason="ticket #46349"),
"xception": pytest.mark.skip(reason="gitlab issue #28"),
"mask_rcnn": pytest.mark.nightly(),
},
"magnitude_sparsity": {
"inception_resnet_v2": pytest.mark.skip(reason="gitlab issue #17"),
"nasnet_mobile": pytest.mark.skip(reason="gitlab issue #18"),
"xception": pytest.mark.skip(reason="gitlab issue #28"),
"mask_rcnn": pytest.mark.nightly(),
},
"filter_pruning": {
"densenet121": pytest.mark.skip(reason="ticket #50604"),
Expand All @@ -214,7 +216,7 @@ def __init__(
"resnet50v2": pytest.mark.skip(reason="Several masks on one weight"),
"mobilenet_v2_slim": pytest.mark.skip(reason="ticket #46349"),
},
"rb_sparsity": {"mobilenet_v2_slim": pytest.mark.skip(reason="ticket #46349")},
"rb_sparsity": {"mobilenet_v2_slim": pytest.mark.skip(reason="ticket #46349"), "mask_rcnn": pytest.mark.nightly()},
}


Expand Down
1 change: 0 additions & 1 deletion tests/tensorflow/test_sota_checkpoints.py
Original file line number Diff line number Diff line change
Expand Up @@ -250,7 +250,6 @@ def metrics_dump_dir(request: FixtureRequest):
return dump_path


@pytest.mark.nightly
class TestSotaCheckpoints:
@pytest.fixture(scope="class")
def collected_data(self, metrics_dump_dir: Path):
Expand Down

0 comments on commit fae793d

Please sign in to comment.