Skip to content

Commit

Permalink
skbase deps
Browse files Browse the repository at this point in the history
  • Loading branch information
fkiraly committed Dec 15, 2024
1 parent b576914 commit 8d74d91
Show file tree
Hide file tree
Showing 24 changed files with 31 additions and 601 deletions.
2 changes: 1 addition & 1 deletion skpro/benchmarking/evaluate.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,9 @@

import numpy as np
import pandas as pd
from skbase.utils.dependencies import _check_soft_dependencies

from skpro.utils.parallel import parallelize
from skpro.utils.validation._dependencies import _check_soft_dependencies


def _split(X, y, C, train, test):
Expand Down
2 changes: 1 addition & 1 deletion skpro/benchmarking/tests/test_evaluate.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
import numpy as np
import pandas as pd
import pytest
from skbase.utils.dependencies import _check_soft_dependencies
from sklearn.datasets import load_diabetes
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import KFold, ShuffleSplit
Expand All @@ -16,7 +17,6 @@
from skpro.metrics import CRPS, EmpiricalCoverage, LogLoss, PinballLoss
from skpro.regression.residual import ResidualDouble
from skpro.tests.test_switch import run_test_module_changed
from skpro.utils.validation._dependencies import _check_soft_dependencies


def _check_evaluate_output(out, cv, y, scoring):
Expand Down
2 changes: 1 addition & 1 deletion skpro/datatypes/_registry.py
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ def scitype_to_mtype(scitype: str, softdeps: str = "exclude"):
return mtypes

if softdeps == "present":
from skpro.utils.validation._dependencies import _check_soft_dependencies
from skbase.utils.dependencies import _check_soft_dependencies

def present(x):
"""Return True if x has satisfied soft dependency or has no soft dep."""
Expand Down
2 changes: 1 addition & 1 deletion skpro/datatypes/_table/_convert.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,11 +32,11 @@

import numpy as np
import pandas as pd
from skbase.utils.dependencies import _check_soft_dependencies

from skpro.datatypes._base import BaseConverter
from skpro.datatypes._convert_utils._convert import _extend_conversions
from skpro.datatypes._table._registry import MTYPE_LIST_TABLE
from skpro.utils.validation._dependencies import _check_soft_dependencies

##############################################################
# methods to convert one machine type to another machine type
Expand Down
2 changes: 1 addition & 1 deletion skpro/datatypes/tests/test_lookup.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
__author__ = ["fkiraly"]

import pytest
from skbase.utils.dependencies import _check_soft_dependencies

from skpro.datatypes._registry import (
MTYPE_REGISTER,
Expand All @@ -11,7 +12,6 @@
scitype_to_mtype,
)
from skpro.tests.test_switch import run_test_module_changed
from skpro.utils.validation._dependencies import _check_soft_dependencies

MTYPE_SCITYPE_PAIRS = [(k[0], k[1]) for k in MTYPE_REGISTER]

Expand Down
2 changes: 1 addition & 1 deletion skpro/datatypes/tests/test_polars.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,11 @@

import pandas as pd
import pytest
from skbase.utils.dependencies import _check_soft_dependencies
from sklearn.datasets import load_diabetes
from sklearn.model_selection import train_test_split

from skpro.tests.test_switch import run_test_module_changed
from skpro.utils.validation._dependencies import _check_soft_dependencies

if _check_soft_dependencies(["polars", "pyarrow"], severity="none"):
import polars as pl
Expand Down
6 changes: 3 additions & 3 deletions skpro/distributions/base/_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,13 +9,13 @@

import numpy as np
import pandas as pd

from skpro.base import BaseObject
from skpro.utils.validation._dependencies import (
from skbase.utils.dependencies import (
_check_estimator_deps,
_check_soft_dependencies,
)

from skpro.base import BaseObject


class BaseDistribution(BaseObject):
"""Base probability distribution."""
Expand Down
2 changes: 1 addition & 1 deletion skpro/distributions/tests/test_proba_basic.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,9 @@
import numpy as np
import pandas as pd
import pytest
from skbase.utils.dependencies import _check_soft_dependencies

from skpro.tests.test_switch import run_test_module_changed
from skpro.utils.validation._dependencies import _check_soft_dependencies


@pytest.mark.skipif(
Expand Down
9 changes: 5 additions & 4 deletions skpro/regression/base/_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,14 +3,15 @@

import numpy as np
import pandas as pd

from skpro.base import BaseEstimator
from skpro.datatypes import check_is_error_msg, check_is_mtype, convert
from skpro.utils.validation._dependencies import (
from skbase.utils.dependencies import (
_check_estimator_deps,
_check_soft_dependencies,
)

from skpro.base import BaseEstimator
from skpro.datatypes import check_is_error_msg, check_is_mtype, convert


# allowed input mtypes
# include mtypes that are core dependencies
ALLOWED_MTYPES = [
Expand Down
3 changes: 2 additions & 1 deletion skpro/regression/bayesian.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,9 @@
# copyright: skpro developers
__author__ = ["meraldoantonio"]

from skbase.utils.dependencies import _check_soft_dependencies

from skpro.regression.base import BaseProbaRegressor
from skpro.utils.validation._dependencies import _check_soft_dependencies


class BayesianLinearRegressor(BaseProbaRegressor):
Expand Down
2 changes: 1 addition & 1 deletion skpro/regression/compose/_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -199,11 +199,11 @@ def get_test_params(cls, parameter_set="default"):
`MyClass(**params)` or `MyClass(**params[i])` creates a valid test instance.
`create_test_instance` uses the first (or only) dictionary in `params`
"""
from skbase.utils.dependencies import _check_estimator_deps
from sklearn.preprocessing import StandardScaler

from skpro.regression.residual import ResidualDouble
from skpro.survival.coxph import CoxPH
from skpro.utils.validation._dependencies import _check_estimator_deps

params = []

Expand Down
2 changes: 1 addition & 1 deletion skpro/regression/online/_dont_refit.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,11 +89,11 @@ def get_test_params(cls, parameter_set="default"):
`MyClass(**params)` or `MyClass(**params[i])` creates a valid test instance.
`create_test_instance` uses the first (or only) dictionary in `params`
"""
from skbase.utils.dependencies import _check_estimator_deps
from sklearn.linear_model import LinearRegression, Ridge

from skpro.regression.residual import ResidualDouble
from skpro.survival.coxph import CoxPH
from skpro.utils.validation._dependencies import _check_estimator_deps

regressor = ResidualDouble(LinearRegression())

Expand Down
2 changes: 1 addition & 1 deletion skpro/regression/online/_refit.py
Original file line number Diff line number Diff line change
Expand Up @@ -164,11 +164,11 @@ def get_test_params(cls, parameter_set="default"):
`MyClass(**params)` or `MyClass(**params[i])` creates a valid test instance.
`create_test_instance` uses the first (or only) dictionary in `params`
"""
from skbase.utils.dependencies import _check_estimator_deps
from sklearn.linear_model import LinearRegression, Ridge

from skpro.regression.residual import ResidualDouble
from skpro.survival.coxph import CoxPH
from skpro.utils.validation._dependencies import _check_estimator_deps

regressor = ResidualDouble(LinearRegression())

Expand Down
2 changes: 1 addition & 1 deletion skpro/regression/online/_refit_every.py
Original file line number Diff line number Diff line change
Expand Up @@ -143,11 +143,11 @@ def get_test_params(cls, parameter_set="default"):
`MyClass(**params)` or `MyClass(**params[i])` creates a valid test instance.
`create_test_instance` uses the first (or only) dictionary in `params`
"""
from skbase.utils.dependencies import _check_estimator_deps
from sklearn.linear_model import LinearRegression, Ridge

from skpro.regression.residual import ResidualDouble
from skpro.survival.coxph import CoxPH
from skpro.utils.validation._dependencies import _check_estimator_deps

regressor = ResidualDouble(LinearRegression())

Expand Down
3 changes: 2 additions & 1 deletion skpro/tests/test_switch.py
Original file line number Diff line number Diff line change
Expand Up @@ -162,9 +162,10 @@ class for which to determine whether it should be tested
If multiple reasons are present, the first one in the above list is returned.
"""
from skbase.utils.dependencies import _check_estimator_deps

from skpro.tests.test_all_estimators import ONLY_CHANGED_MODULES
from skpro.utils.git_diff import get_packages_with_changed_specs, is_class_changed
from skpro.utils.validation._dependencies import _check_estimator_deps

PACKAGE_REQ_CHANGED = get_packages_with_changed_specs()

Expand Down
3 changes: 2 additions & 1 deletion skpro/tests/tests/test_test_utils.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
"""Tests for the test utilities."""

from skbase.utils.dependencies import _check_estimator_deps

from skpro.tests._config import EXCLUDE_ESTIMATORS
from skpro.tests.test_switch import run_test_for_class
from skpro.utils.validation._dependencies import _check_estimator_deps


def test_exclude_estimators():
Expand Down
2 changes: 1 addition & 1 deletion skpro/utils/_maint/tests/test_show_versions.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
"""Tests for the show_versions utility."""
import pathlib
import uuid
from skbase.utils.dependencies import _check_soft_dependencies

from skpro.utils._maint._show_versions import (
DEFAULT_DEPS_TO_SHOW,
_get_deps_info,
show_versions,
)
from skpro.utils.validation._dependencies import _check_soft_dependencies


def test_show_versions_runs():
Expand Down
2 changes: 1 addition & 1 deletion skpro/utils/deep_equals/_deep_equals.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
"""
from skbase.utils.deep_equals._common import _make_ret
from skbase.utils.deep_equals._deep_equals import deep_equals as _deep_equals
from skbase.utils.dependencies import _check_soft_dependencies

from skpro.utils.validation._dependencies import _check_soft_dependencies

__author__ = ["fkiraly"]
__all__ = ["deep_equals"]
Expand Down
2 changes: 1 addition & 1 deletion skpro/utils/estimator_checks.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
__author__ = ["fkiraly"]
__all__ = ["check_estimator"]

from skpro.utils.validation._dependencies import _check_soft_dependencies
from skbase.utils.dependencies import _check_soft_dependencies


def check_estimator(
Expand Down
2 changes: 1 addition & 1 deletion skpro/utils/parallel.py
Original file line number Diff line number Diff line change
Expand Up @@ -155,7 +155,7 @@ def _get_parallel_test_fixtures():
Returns a list of parameter fixtures, where each fixture
is a dict with keys "backend" and "backend_params".
"""
from skpro.utils.validation._dependencies import _check_soft_dependencies
from skbase.utils.dependencies import _check_soft_dependencies

fixtures = []

Expand Down
2 changes: 1 addition & 1 deletion skpro/utils/plotting.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
import numpy as np
import pandas as pd

from skpro.utils.validation._dependencies import _check_soft_dependencies
from skbase.utils.dependencies import _check_soft_dependencies

__authors__ = ["fkiraly", "frthjf"]

Expand Down
2 changes: 1 addition & 1 deletion skpro/utils/tests/test_plots.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,9 @@
# copyright: skpro developers, BSD-3-Clause License (see LICENSE file)

import pytest
from skbase.utils.dependencies import _check_soft_dependencies

from skpro.tests.test_switch import run_test_module_changed
from skpro.utils.validation._dependencies import _check_soft_dependencies


@pytest.mark.skipif(
Expand Down
14 changes: 0 additions & 14 deletions skpro/utils/validation/__init__.py

This file was deleted.

Loading

0 comments on commit 8d74d91

Please sign in to comment.