Skip to content

Commit 2624df2

Browse files
committed
Do not fail due to onnx issue in examples
1 parent 06e5f33 commit 2624df2

File tree

1 file changed

+1
-15
lines changed

1 file changed

+1
-15
lines changed

tests/docs/test_examples.py

Lines changed: 1 addition & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,6 @@
88

99
from baybe._optional.info import CHEM_INSTALLED, ONNX_INSTALLED
1010

11-
from ..conftest import _onnx_issue
12-
1311
# Run these tests in reduced settings
1412
_SMOKE_TEST_CACHE = os.environ.get("SMOKE_TEST", None)
1513
os.environ["SMOKE_TEST"] = "true"
@@ -22,19 +20,7 @@
2220
@pytest.mark.skipif(
2321
not (CHEM_INSTALLED and ONNX_INSTALLED), reason="skipped for core tests"
2422
)
25-
@pytest.mark.parametrize(
26-
"example",
27-
[
28-
pytest.param(
29-
p,
30-
marks=[pytest.mark.xfail(reason=_onnx_issue, strict=True)]
31-
if "custom_pretrained" in p
32-
else [],
33-
)
34-
for p in paths
35-
],
36-
ids=paths,
37-
)
23+
@pytest.mark.parametrize("example", paths, ids=paths)
3824
def test_example(example: str):
3925
"""Test an individual example by running it.
4026

0 commit comments

Comments
 (0)