diff --git a/pyproject.toml b/pyproject.toml index 101cfb7a..88814539 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -71,8 +71,8 @@ tutorials = [ "jupyter", "seaborn", "matplotlib", - "datasets>=2.8.0, - "scikit-learn">=1.1.0, + "datasets>=2.8.0", + "scikit-learn>=1.1.0", ] [project.readme] diff --git a/src/textdescriptives/sklearn_featurizer.py b/src/textdescriptives/sklearn_featurizer.py index aea5f841..16d2233f 100644 --- a/src/textdescriptives/sklearn_featurizer.py +++ b/src/textdescriptives/sklearn_featurizer.py @@ -1,6 +1,5 @@ from typing import Iterable, List, Optional -import numpy as np import pandas as pd from sklearn.base import BaseEstimator, TransformerMixin from wasabi import msg @@ -106,5 +105,3 @@ def get_feature_names_out(self, input_features=None) -> List[str]: """Get the names of the extracted features. input_features is only present for API compatibility with sklearn.""" return self.feature_names - -