fixes plotly#805

cldougl · cldougl · commit c1e9bd24e10e · 2017-08-04T16:10:15.000-04:00
diff --git a/plotly/figure_factory/_dendrogram.py b/plotly/figure_factory/_dendrogram.py
@@ -87,8 +87,7 @@ def create_dendrogram(X, orientation="bottom", labels=None,
     dendrogram = _Dendrogram(X, orientation, labels, colorscale,
                              distfun=distfun, linkagefun=linkagefun)
 
-    return {'layout': dendrogram.layout,
-            'data': dendrogram.data}
+    return graph_objs.Figure(data=dendrogram.data, layout=dendrogram.layout)
 
 
 class _Dendrogram(object):
diff --git a/plotly/figure_factory/_gantt.py b/plotly/figure_factory/_gantt.py
@@ -4,6 +4,7 @@
 
 from plotly import exceptions, optional_imports
 from plotly.figure_factory import utils
+from plotly.graph_objs import graph_objs
 
 pd = optional_imports.get_module('pandas')
 
@@ -170,7 +171,7 @@ def gantt(chart, colors, title, bar_width, showgrid_x, showgrid_y, height,
     )
     layout['shapes'] = tasks
 
-    fig = dict(data=data, layout=layout)
+    fig = graph_objs.Figure(data=data, layout=layout)
     return fig
 
 
