Merge pull request #3714 from plotly/more_types

missed some type annotations

Author: nicolaskruchten

Diff for: packages/python/plotly/codegen/figure.py

@@ -644,6 +644,7 @@ def add_{method_prefix}{singular_name}(self"""
             node.child_datatypes,
             prepend_extras=["arg"],
             append_extras=["row", "col", "secondary_y", "exclude_empty_subplots"],
+            output_type=fig_classname,
         )
 
         prepend_extras = [

Diff for: packages/python/plotly/plotly/graph_objs/_figure.py

@@ -21785,7 +21785,7 @@ def add_annotation(
         secondary_y=None,
         exclude_empty_subplots=None,
         **kwargs,
-    ):
+    ) -> "Figure":
         """
         Create and add a new annotation to the figure's layout
 
@@ -22334,7 +22334,7 @@ def add_layout_image(
         secondary_y=None,
         exclude_empty_subplots=None,
         **kwargs,
-    ):
+    ) -> "Figure":
         """
         Create and add a new image to the figure's layout
 
@@ -22665,7 +22665,7 @@ def add_shape(
         secondary_y=None,
         exclude_empty_subplots=None,
         **kwargs,
-    ):
+    ) -> "Figure":
         """
         Create and add a new shape to the figure's layout
 

Diff for: packages/python/plotly/plotly/graph_objs/_figurewidget.py

@@ -21791,7 +21791,7 @@ def add_annotation(
         secondary_y=None,
         exclude_empty_subplots=None,
         **kwargs,
-    ):
+    ) -> "FigureWidget":
         """
         Create and add a new annotation to the figure's layout
 
@@ -22340,7 +22340,7 @@ def add_layout_image(
         secondary_y=None,
         exclude_empty_subplots=None,
         **kwargs,
-    ):
+    ) -> "FigureWidget":
         """
         Create and add a new image to the figure's layout
 
@@ -22671,7 +22671,7 @@ def add_shape(
         secondary_y=None,
         exclude_empty_subplots=None,
         **kwargs,
-    ):
+    ) -> "FigureWidget":
         """
         Create and add a new shape to the figure's layout