Skip to content

Commit

Permalink
Formatting.
Browse files Browse the repository at this point in the history
  • Loading branch information
umaannamalai committed Sep 19, 2023
1 parent 152807e commit 8e7dda3
Show file tree
Hide file tree
Showing 9 changed files with 9 additions and 0 deletions.
1 change: 1 addition & 0 deletions tests/mlmodel_sklearn/test_ensemble_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@

SKLEARN_VERSION = tuple(map(int, get_package_version("sklearn").split(".")))


@pytest.mark.parametrize(
"ensemble_model_name",
[
Expand Down
1 change: 1 addition & 0 deletions tests/mlmodel_sklearn/test_feature_selection_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@

SKLEARN_VERSION = tuple(map(int, get_package_version("sklearn").split(".")))


@pytest.mark.parametrize(
"feature_selection_model_name",
[
Expand Down
1 change: 1 addition & 0 deletions tests/mlmodel_sklearn/test_gaussian_process_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
from newrelic.api.background_task import background_task
from newrelic.packages import six


@pytest.mark.parametrize(
"gaussian_process_model_name",
[
Expand Down
1 change: 1 addition & 0 deletions tests/mlmodel_sklearn/test_kernel_ridge_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
from newrelic.api.background_task import background_task
from newrelic.packages import six


@pytest.mark.parametrize(
"kernel_ridge_model_name",
[
Expand Down
1 change: 1 addition & 0 deletions tests/mlmodel_sklearn/test_linear_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@

SKLEARN_VERSION = tuple(map(int, get_package_version("sklearn").split(".")))


@pytest.mark.parametrize(
"linear_model_name",
[
Expand Down
1 change: 1 addition & 0 deletions tests/mlmodel_sklearn/test_metric_scorers.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@
from newrelic.api.background_task import background_task
from newrelic.hooks.mlmodel_sklearn import PredictReturnTypeProxy


@pytest.mark.parametrize(
"metric_scorer_name",
(
Expand Down
1 change: 1 addition & 0 deletions tests/mlmodel_sklearn/test_ml_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -137,6 +137,7 @@ def _test():
),
]


@reset_core_stats_engine()
def test_custom_model_int_list_with_metadata():
@validate_ml_event_count(count=1)
Expand Down
1 change: 1 addition & 0 deletions tests/mlmodel_sklearn/test_model_selection_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@
from newrelic.api.background_task import background_task
from newrelic.packages import six


@pytest.mark.parametrize(
"model_selection_model_name",
[
Expand Down
1 change: 1 addition & 0 deletions tests/mlmodel_sklearn/test_multiclass_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@
from newrelic.api.background_task import background_task
from newrelic.packages import six


@pytest.mark.parametrize(
"multiclass_model_name",
[
Expand Down

0 comments on commit 8e7dda3

Please sign in to comment.