Fix dependencies

Signed-off-by: Beat Buesser <[email protected]>

Author: beat-buesser

art/evaluations/security_curve/security_curve.py

@@ -165,7 +165,7 @@ def plot(self) -> None:  # pragma: no cover
         from matplotlib import pyplot as plt
 
         plt.plot(self.eps_list, self.accuracy_adv_list, label="adversarial", marker="o")
-        plt.plot([self.eps_list[0], self.eps_list[-1]], [self.accuracy, self.accuracy], linestyle="--", label="benign")
+        plt.plot([self.eps_list[0], self.eps_list[-1]], [self.accuracy, self.accuracy], linestyle="--", label="benign")  # type: ignore
         plt.legend()
         plt.xlabel("Attack budget eps")
         plt.ylabel("Accuracy")

art/visualization.py

@@ -150,7 +150,7 @@ def plot_3d(
     for i, coord in enumerate(points):
         try:
             color_point = labels[i]
-            axis.scatter3D(coord[0], coord[1], coord[2], color=colors[color_point])
+            axis.scatter3D(coord[0], coord[1], coord[2], color=colors[color_point])  # type: ignore
         except IndexError:
             raise ValueError(
                 "Labels outside the range. Should start from zero and be sequential there after"