mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-29 23:06:41 +00:00
d433aa2459
* checkin transformers pipeline * add docker requirements * only trigger linux cpu * temp remove tf instalation due to numpy version conflicts * test numpy>=1.7 * revert numpy and disable transformers * add coloredlogs * enable shape_infer_helper and install transformers when needed * pip3? * testtest * enable more tets * line too long * remove pytorch1.4 test and added back some onnx files * add tests * copy dir * disable 2 teests * trim lines * add missing onnx * fix type * fix version conflicts * install psutil * change file path * mfix path * remove cached files * add back attention fusion test * labeled the shape infer test as slow * fix * enable tf2onnx test and enable pytest * refactor path * fix typo * add cwd
26 lines
902 B
Python
26 lines
902 B
Python
# -------------------------------------------------------------------------
|
|
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License. See License.txt in the project root for
|
|
# license information.
|
|
# --------------------------------------------------------------------------
|
|
"""Configuration for pytest."""
|
|
|
|
import pytest
|
|
|
|
|
|
def pytest_addoption(parser):
|
|
parser.addoption("--slow", action="store_true", default=False, help="run slow tests")
|
|
|
|
|
|
def pytest_configure(config):
|
|
config.addinivalue_line("markers", "slow: mark test as slow to run")
|
|
|
|
|
|
def pytest_collection_modifyitems(config, items):
|
|
if config.getoption("--slow"):
|
|
# --slow given: do not skip slow tests
|
|
return
|
|
skip_slow = pytest.mark.skip(reason="need --slow option to run")
|
|
for item in items:
|
|
if "slow" in item.keywords:
|
|
item.add_marker(skip_slow)
|