152 ruff apply unsafe code fixes for code basis (#153)

Author: c-w-feldmann

README.md

@@ -122,7 +122,7 @@ physchem_matrix
 #                [88.065, 20.23 ,  1.   ]])
 ```
 
-MolPipeline provides further features and descriptors from RDKit, 
+MolPipeline provides further features and descriptors from RDKit,
 for example Morgan (binary/count) fingerprints and MACCS keys.
 See the [04_feature_calculation notebook](notebooks/04_feature_calculation.ipynb) for more examples.
 

molpipeline/estimators/chemprop/component_wrapper.py

@@ -31,9 +31,7 @@
 from chemprop.nn.predictors import MveFFN as _MveFFN
 from chemprop.nn.predictors import RegressionFFN as _RegressionFFN
 from chemprop.nn.predictors import SpectralFFN as _SpectralFFN
-from chemprop.nn.predictors import (
-    _FFNPredictorBase as _Predictor,  # pylint: disable=protected-access
-)
+from chemprop.nn.predictors import _FFNPredictorBase as _Predictor
 from chemprop.nn.transforms import UnscaleTransform
 from chemprop.nn.utils import Activation, get_activation_function
 from sklearn.base import BaseEstimator
@@ -477,7 +475,7 @@ def reinitialize_network(self) -> Self:
             # pylint: disable=protected-access
             self.metrics = [self.predictor._T_default_metric, self.criterion]
         else:
-            self.metrics = list(self.metric_list) + [self.criterion]
+            self.metrics = [*list(self.metric_list), self.criterion]
 
         return self
 

molpipeline/estimators/chemprop/featurizer_wrapper/graph_wrapper.py

@@ -20,7 +20,8 @@ class SimpleMoleculeMolGraphFeaturizer(_SimpleMoleculeMolGraphFeaturizer):
     extra_bond_fdim: InitVar[int]
 
     def get_params(
-        self, deep: bool = True  # pylint: disable=unused-argument
+        self,
+        deep: bool = True,  # pylint: disable=unused-argument
     ) -> dict[str, InitVar[int]]:
         """Get parameters for the featurizer.
 

molpipeline/estimators/chemprop/models.py

@@ -149,7 +149,8 @@ def __sklearn_tags__(self) -> Tags:
         return tags
 
     def _predict(
-        self, X: MoleculeDataset  # pylint: disable=invalid-name
+        self,
+        X: MoleculeDataset,  # pylint: disable=invalid-name
     ) -> npt.NDArray[np.float64]:
         """Predict the labels.
 
@@ -207,7 +208,8 @@ def fit(
         return super().fit(X, y)
 
     def predict(
-        self, X: MoleculeDataset  # pylint: disable=invalid-name
+        self,
+        X: MoleculeDataset,  # pylint: disable=invalid-name
     ) -> npt.NDArray[np.float64]:
         """Predict the output.
 
@@ -234,7 +236,8 @@ def predict(
 
     @available_if(_is_classifier)
     def predict_proba(
-        self, X: MoleculeDataset  # pylint: disable=invalid-name
+        self,
+        X: MoleculeDataset,  # pylint: disable=invalid-name
     ) -> npt.NDArray[np.float64]:
         """Predict the probabilities.
 

molpipeline/estimators/chemprop/neural_fingerprint.py

@@ -85,7 +85,8 @@ def fit(
         return super().fit(X, y)
 
     def transform(
-        self, X: MoleculeDataset  # pylint: disable=invalid-name
+        self,
+        X: MoleculeDataset,  # pylint: disable=invalid-name
     ) -> npt.NDArray[np.float64]:
         """Transform the input.
 

molpipeline/estimators/similarity_transformation.py

@@ -5,7 +5,7 @@
 from typing import Any
 
 try:
-    from typing import Self  # pylint: disable=no-name-in-module
+    from typing import Self
 except ImportError:
     from typing_extensions import Self
 
@@ -92,7 +92,8 @@ def fit(
         return self
 
     def transform(
-        self, X: npt.NDArray[np.float64] | csr_matrix  # pylint: disable=invalid-name
+        self,
+        X: npt.NDArray[np.float64] | csr_matrix,  # pylint: disable=invalid-name
     ) -> npt.NDArray[np.float64]:
         """Transform the data.
 
@@ -112,7 +113,7 @@ def transform(
 
     def fit_transform(
         self,
-        X: npt.NDArray[np.float64] | csr_matrix,  # pylint: disable=invalid-name
+        X: npt.NDArray[np.float64] | csr_matrix,
         y: npt.NDArray[np.float64] | None = None,
         **fit_params: Any,
     ) -> npt.NDArray[np.float64]:

molpipeline/experimental/explainability/explainer.py

@@ -162,7 +162,9 @@ class AbstractSHAPExplainer(abc.ABC):  # pylint: disable=too-few-public-methods
 
     @abc.abstractmethod
     def explain(
-        self, X: Any, **kwargs: Any  # pylint: disable=invalid-name
+        self,
+        X: Any,  # pylint: disable=invalid-name
+        **kwargs: Any,
     ) -> list[SHAPFeatureExplanation | SHAPFeatureAndAtomExplanation]:
         """Explain the predictions for the input data.
 
@@ -282,7 +284,6 @@ def explain(
             SHAPFeatureExplanation | SHAPFeatureAndAtomExplanation
         ] = []
         for input_sample in X:
-
             input_sample = [input_sample]
 
             # get predictions

molpipeline/experimental/explainability/visualization/visualization.py

@@ -358,7 +358,7 @@ def _add_shap_present_absent_features_text(
     )
 
     delta = 0.05
-    offset = offset + 0.0165
+    offset += 0.0165
     fig.text(offset, 0.13, "prediction =", ha="center", fontsize=10)
     fig.text(
         offset + 2 * delta,
@@ -558,7 +558,6 @@ def structure_heatmap_shap(  # pylint: disable=too-many-locals
     )
 
     with plt.ioff():
-
         drawer, _, _, normalizer, color_map = _structure_heatmap(
             explanation.molecule,
             explanation.atom_weights,

molpipeline/metrics/ignore_error_scorer.py

@@ -9,7 +9,7 @@
 import pandas as pd
 from loguru import logger
 from sklearn import metrics
-from sklearn.metrics._scorer import _BaseScorer  # pylint: disable=protected-access
+from sklearn.metrics._scorer import _BaseScorer
 
 
 def ignored_value_scorer(
@@ -80,10 +80,10 @@ def newscore(
             )
         y_true_ = np.copy(np.array(y_true)[all_retained])
         y_pred_ = np.array(np.array(y_pred)[all_retained].tolist())
-        _kwargs = dict(kwargs)
-        if "sample_weight" in _kwargs and _kwargs["sample_weight"] is not None:
-            _kwargs["sample_weight"] = _kwargs["sample_weight"][all_retained]
-        return score_func(y_true_, y_pred_, **_kwargs)
+        kwargs_ = dict(kwargs)
+        if "sample_weight" in kwargs_ and kwargs_["sample_weight"] is not None:
+            kwargs_["sample_weight"] = kwargs_["sample_weight"][all_retained]
+        return score_func(y_true_, y_pred_, **kwargs_)
 
     return metrics.make_scorer(
         newscore, response_method=response_method, **scorer_kwargs

molpipeline/mol2any/mol2chemprop.py

@@ -22,6 +22,7 @@
     warnings.warn(
         "chemprop not installed. MolToChemprop will not work.",
         ImportWarning,
+        stacklevel=2,
     )