mypy passing

Author: Julien Roussel

qolmat/imputations/rpca/rpca_noisy.py

@@ -389,8 +389,7 @@ def cost_function(
         tau: float,
         lam: float,
     ):
-
-        temporal_norm = 0
+        temporal_norm: float = 0
         if len(self.list_etas) > 0:
             # matrices for temporal correlation
             H = [
@@ -402,7 +401,7 @@ def cost_function(
                     temporal_norm += eta * np.sum(np.abs(H_matrix @ low_rank))
             elif self.norm == "L2":
                 for eta, H_matrix in zip(self.list_etas, H):
-                    temporal_norm += eta * np.linalg.norm(low_rank @ H_matrix, "fro")
+                    temporal_norm += eta * float(np.linalg.norm(low_rank @ H_matrix, "fro"))
         anomalies_norm = np.sum(np.abs(anomalies * Omega))
         cost = (
             1 / 2 * ((Omega * (observations - low_rank - anomalies)) ** 2).sum()

tests/utils/test_plot.py

@@ -1,4 +1,4 @@
-from typing import List, Tuple
+from typing import Any, List, Tuple
 import matplotlib as mpl
 import matplotlib.pyplot as plt
 import numpy as np
@@ -45,7 +45,7 @@ def test_utils_plot_plot_matrices(list_matrices: List[np.ndarray], mocker: Mocke
 
 
 @pytest.mark.parametrize("list_signals", [list_signals])
-def test_utils_plot_plot_signal(list_signals: List[np.ndarray], mocker: MockerFixture) -> None:
+def test_utils_plot_plot_signal(list_signals: List[List[Any]], mocker: MockerFixture) -> None:
     mocker.patch("matplotlib.pyplot.savefig")
     mocker.patch("matplotlib.pyplot.show")
     plot.plot_signal(list_signals=list_signals, ylabel="ylabel", title="title")