test: Fix typing issues (#384)

* Avoid reassignment in update_gradient_coordinate
* Fix wrong import of Figure

Author: ValerianRey

tests/plots/_utils.py

@@ -1,8 +1,7 @@
 import numpy as np
 import torch
-from dash.html import Figure
 from plotly import graph_objects as go
-from plotly.graph_objs import Scatter
+from plotly.graph_objs import Figure, Scatter
 
 from torchjd.aggregation import Aggregator
 

tests/plots/interactive_plotter.py

@@ -6,7 +6,7 @@
 import numpy as np
 import torch
 from dash import Dash, Input, Output, callback, dcc, html
-from dash.html import Figure
+from plotly.graph_objs import Figure
 from plots._utils import Plotter, angle_to_coord, coord_to_angle
 
 from torchjd.aggregation import (
@@ -121,11 +121,11 @@ def update_seed(value: int) -> Figure:
         prevent_initial_call=True,
     )
     def update_gradient_coordinate(*values) -> Figure:
-        values = [float(value) for value in values]
+        values_ = [float(value) for value in values]
 
-        for j in range(len(values) // 2):
-            angle = values[2 * j]
-            r = values[2 * j + 1]
+        for j in range(len(values_) // 2):
+            angle = values_[2 * j]
+            r = values_[2 * j + 1]
             x, y = angle_to_coord(angle, r)
             plotter.matrix[j, 0] = x
             plotter.matrix[j, 1] = y