From 1afde5b5ed2c2cd3e0f4c195cd5d688349613951 Mon Sep 17 00:00:00 2001 From: Oleksii Kachaiev Date: Mon, 5 Feb 2024 15:28:40 +0100 Subject: [PATCH] Fix flake8 for utils and tests (#73) --- skada/_utils.py | 3 ++- skada/tests/test_utils.py | 10 +++++----- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/skada/_utils.py b/skada/_utils.py index 8da27075..60d6c2e1 100644 --- a/skada/_utils.py +++ b/skada/_utils.py @@ -17,6 +17,7 @@ ) from sklearn.utils.multiclass import type_of_target + _logger = logging.getLogger('skada') _logger.setLevel(logging.DEBUG) @@ -93,7 +94,7 @@ def _find_y_type(y): """ # We need to check for this case first because - # type_of_target() doesnt handle nan values + # type_of_target() doesn't handle nan values if np.any(np.isnan(y)): if y.ndim != 1: raise ValueError("For a regression task, " diff --git a/skada/tests/test_utils.py b/skada/tests/test_utils.py index 1b99d4c2..3ca5b43a 100644 --- a/skada/tests/test_utils.py +++ b/skada/tests/test_utils.py @@ -5,15 +5,15 @@ import pytest import numpy as np -from skada.datasets import ( - make_dataset_from_moons_distribution -) +from skada.datasets import make_dataset_from_moons_distribution from skada.utils import ( check_X_y_domain, check_X_domain, - extract_source_indices + extract_source_indices, + source_target_split, + source_target_merge ) -from skada.utils import source_target_split, source_target_merge +from skada.utils import source_target_split from skada._utils import _check_y_masking