Skip to content

Commit

Permalink
style fix
Browse files Browse the repository at this point in the history
  • Loading branch information
bstrzele committed Jun 3, 2024
1 parent 784904e commit 1ef1211
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 3 deletions.
2 changes: 1 addition & 1 deletion tests/configs/dynamic_gpu.pbtxt
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ output: [
dims: [ 1, 1, -1, 7]
}
]
parameters: [
parameters: [
{
key: "RESHAPE_IO_LAYERS"
value: {
Expand Down
7 changes: 6 additions & 1 deletion tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ def pytest_addoption(parser):
parser.addoption("--model-cache", action="store")
parser.addoption("--image", action="store")


def pytest_configure(config):
config.addinivalue_line(
"markers",
Expand All @@ -21,4 +22,8 @@ def pytest_collection_modifyitems(config, items):
for item in items:
if "gpu" in item.keywords:
if not config.getoption("--gpu"):
item.add_marker(pytest.mark.skip("Test requires --gpu flag to be set and Intel GPU device on the host machine"))
item.add_marker(
pytest.mark.skip(
"Test requires --gpu flag to be set and Intel GPU device on the host machine"
)
)
2 changes: 1 addition & 1 deletion tests/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
pytest
tritonclient[all]
requests
tritonclient[all]

0 comments on commit 1ef1211

Please sign in to comment.