Skip to content

Commit 2df0e5d

Browse files
Merge pull request #76 from XXII-AI/fix/pytest_regressions
Do not import pytest regression if regression extra is not installed
2 parents bd614a8 + d6e7cb1 commit 2df0e5d

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

lours/utils/regression_testing.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44

55
import pytest
66
from imageio import imread
7-
from pytest_regressions.common import perform_regression_check
87

98
from lours.dataset import Dataset, from_parquet
109

@@ -125,6 +124,14 @@ def check(
125124
self.dump_fn, dataset, with_image_checksum=check_images
126125
)
127126

127+
try:
128+
from pytest_regressions.common import perform_regression_check
129+
except ImportError as e:
130+
raise ImportError(
131+
"Pytest regression is not installed, reinstall Lours with "
132+
"'regression' extra dependency"
133+
) from e
134+
128135
perform_regression_check(
129136
datadir=self.datadir,
130137
original_datadir=self.original_datadir,

0 commit comments

Comments
 (0)