linting

basf · Jan 14, 2025 · dcf95c2 · dcf95c2
1 parent 3bc87e3
commit dcf95c2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/molpipeline/experimental/explainability/explainer.py b/molpipeline/experimental/explainability/explainer.py
@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 import abc
-from typing import Any, TypeAlias, Callable
+from typing import Any, Callable, TypeAlias
 
 import numpy as np
 import numpy.typing as npt

diff --git a/molpipeline/experimental/explainability/visualization/heatmaps.py b/molpipeline/experimental/explainability/visualization/heatmaps.py
@@ -176,9 +176,7 @@ def __init__(
                 Callable[[npt.NDArray[np.float64]], npt.NDArray[np.float64]]
             ] = function_list
         else:
-            self.function_list: list[
-                Callable[[npt.NDArray[np.float64]], npt.NDArray[np.float64]]
-            ] = []
+            self.function_list = []
         self.values = np.zeros((self.x_res, self.y_res))
 
     def add_function(