Skip to content

Commit 4039d06

Browse files
authored
Merge branch 'main' into drop-none-attrs
2 parents 111a2c4 + 0f89d99 commit 4039d06

12 files changed

+25
-26
lines changed

tests/mlmodel_sklearn/test_cluster_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,10 @@
1919
)
2020

2121
from newrelic.api.background_task import background_task
22-
from newrelic.common.package_version_utils import get_package_version
22+
from newrelic.common.package_version_utils import get_package_version_tuple
2323
from newrelic.packages import six
2424

25-
SKLEARN_VERSION = tuple(map(int, get_package_version("sklearn").split(".")))
25+
SKLEARN_VERSION = get_package_version_tuple("sklearn")
2626

2727

2828
@pytest.mark.parametrize(

tests/mlmodel_sklearn/test_dummy_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,10 @@
1919
)
2020

2121
from newrelic.api.background_task import background_task
22-
from newrelic.common.package_version_utils import get_package_version
22+
from newrelic.common.package_version_utils import get_package_version_tuple
2323
from newrelic.packages import six
2424

25-
SKLEARN_VERSION = tuple(map(int, get_package_version("sklearn").split(".")))
25+
SKLEARN_VERSION = get_package_version_tuple("sklearn")
2626

2727

2828
@pytest.mark.parametrize(

tests/mlmodel_sklearn/test_ensemble_models.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,10 @@
1919
)
2020

2121
from newrelic.api.background_task import background_task
22-
from newrelic.common.package_version_utils import get_package_version
22+
from newrelic.common.package_version_utils import get_package_version_tuple
2323
from newrelic.packages import six
2424

25-
SKLEARN_VERSION = tuple(map(int, get_package_version("sklearn").split(".")))
25+
SKLEARN_VERSION = get_package_version_tuple("sklearn")
2626

2727

2828
@pytest.mark.parametrize(
@@ -197,7 +197,6 @@ def _test():
197197
_test()
198198

199199

200-
201200
@pytest.mark.skipif(SKLEARN_VERSION < (1, 1, 0), reason="Requires sklearn >= 1.1")
202201
@pytest.mark.parametrize(
203202
"ensemble_model_name",

tests/mlmodel_sklearn/test_feature_selection_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,10 @@
1919
)
2020

2121
from newrelic.api.background_task import background_task
22-
from newrelic.common.package_version_utils import get_package_version
22+
from newrelic.common.package_version_utils import get_package_version_tuple
2323
from newrelic.packages import six
2424

25-
SKLEARN_VERSION = tuple(map(int, get_package_version("sklearn").split(".")))
25+
SKLEARN_VERSION = get_package_version_tuple("sklearn")
2626

2727

2828
@pytest.mark.parametrize(

tests/mlmodel_sklearn/test_linear_models.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,11 @@
1818
)
1919

2020
from newrelic.api.background_task import background_task
21-
from newrelic.common.package_version_utils import get_package_version
21+
from newrelic.common.package_version_utils import get_package_version_tuple
2222
from newrelic.packages import six
2323

24-
SKLEARN_VERSION = tuple(map(int, get_package_version("sklearn").split(".")))
25-
SCIPY_VERSION = tuple(map(int, get_package_version("scipy").split(".")))
24+
SKLEARN_VERSION = get_package_version_tuple("sklearn")
25+
SCIPY_VERSION = get_package_version_tuple("scipy")
2626

2727

2828
@pytest.mark.parametrize(

tests/mlmodel_sklearn/test_multioutput_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,10 @@
2020
)
2121

2222
from newrelic.api.background_task import background_task
23-
from newrelic.common.package_version_utils import get_package_version
23+
from newrelic.common.package_version_utils import get_package_version_tuple
2424
from newrelic.packages import six
2525

26-
SKLEARN_VERSION = tuple(map(int, get_package_version("sklearn").split(".")))
26+
SKLEARN_VERSION = get_package_version_tuple("sklearn")
2727

2828

2929
# Python 2 will not allow instantiation of abstract class

tests/mlmodel_sklearn/test_naive_bayes_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,10 @@
1919
)
2020

2121
from newrelic.api.background_task import background_task
22-
from newrelic.common.package_version_utils import get_package_version
22+
from newrelic.common.package_version_utils import get_package_version_tuple
2323
from newrelic.packages import six
2424

25-
SKLEARN_VERSION = tuple(map(int, get_package_version("sklearn").split(".")))
25+
SKLEARN_VERSION = get_package_version_tuple("sklearn")
2626

2727

2828
@pytest.mark.skipif(SKLEARN_VERSION < (1, 0, 0), reason="Requires sklearn >= 1.0")

tests/mlmodel_sklearn/test_neighbors_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,10 @@
1919
)
2020

2121
from newrelic.api.background_task import background_task
22-
from newrelic.common.package_version_utils import get_package_version
22+
from newrelic.common.package_version_utils import get_package_version_tuple
2323
from newrelic.packages import six
2424

25-
SKLEARN_VERSION = tuple(map(int, get_package_version("sklearn").split(".")))
25+
SKLEARN_VERSION = get_package_version_tuple("sklearn")
2626

2727

2828
@pytest.mark.parametrize(

tests/mlmodel_sklearn/test_neural_network_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,10 +18,10 @@
1818
)
1919

2020
from newrelic.api.background_task import background_task
21-
from newrelic.common.package_version_utils import get_package_version
21+
from newrelic.common.package_version_utils import get_package_version_tuple
2222
from newrelic.packages import six
2323

24-
SKLEARN_VERSION = tuple(map(int, get_package_version("sklearn").split(".")))
24+
SKLEARN_VERSION = get_package_version_tuple("sklearn")
2525

2626

2727
@pytest.mark.parametrize(

tests/mlmodel_sklearn/test_pipeline_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,10 @@
2121
)
2222

2323
from newrelic.api.background_task import background_task
24-
from newrelic.common.package_version_utils import get_package_version
24+
from newrelic.common.package_version_utils import get_package_version_tuple
2525
from newrelic.packages import six
2626

27-
SKLEARN_VERSION = tuple(map(int, get_package_version("sklearn").split(".")))
27+
SKLEARN_VERSION = get_package_version_tuple("sklearn")
2828

2929

3030
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)