Fix some mypy issues

Author: r3kste

src/optimagic/visualization/backends.py

@@ -428,12 +428,13 @@ def _line_plot_bokeh(
     subplot: Any | None = None,
 ) -> "bokeh.plotting.figure":
     from bokeh.io import curdoc
-    from bokeh.models import Legend, LegendItem, Scatter, Span
+    from bokeh.models import Scatter
+    from bokeh.models.annotations import Legend, LegendItem, Span
     from bokeh.plotting import figure
 
     if template is None:
         template = "light_minimal"
-    curdoc().theme = template
+    curdoc().theme = template  # type: ignore[assignment]
 
     if subplot is not None:
         p = subplot
@@ -580,7 +581,7 @@ def _grid_line_plot_bokeh(
             subplot_row.append(p)
         plots.append(subplot_row)
 
-    grid = gridplot(
+    grid = gridplot(  # type: ignore[call-overload]
         plots,
         height=height // n_rows if height else None,
         width=width // n_cols if width else None,