dev-dsp
diff --git a/‎plotly-package/plotly/figure_factory/_county_choropleth.py
Copy file name to clipboardExpand all lines: plotly-package/plotly/figure_factory/_county_choropleth.py
+2-1Lines changed: 2 additions & 1 deletion b/‎plotly-package/plotly/figure_factory/_county_choropleth.py
Copy file name to clipboardExpand all lines: plotly-package/plotly/figure_factory/_county_choropleth.py
+2-1Lines changed: 2 additions & 1 deletion
diff --git a/‎plotly-package/plotly/tests/test_optional/test_figure_factory/test_figure_factory.py
Copy file name to clipboardExpand all lines: plotly-package/plotly/tests/test_optional/test_figure_factory/test_figure_factory.py
+2-2Lines changed: 2 additions & 2 deletions b/‎plotly-package/plotly/tests/test_optional/test_figure_factory/test_figure_factory.py
Copy file name to clipboardExpand all lines: plotly-package/plotly/tests/test_optional/test_figure_factory/test_figure_factory.py
+2-2Lines changed: 2 additions & 2 deletions
@@ -11,6 +11,7 @@
 import plotly.colors as clrs
 from plotly.figure_factory import utils
 from plotly.exceptions import PlotlyError
+import plotly.graph_objs as go
 
 pd.options.mode.chained_assignment = None
 
@@ -950,4 +951,4 @@ def create_choropleth(fips, values, scope=['usa'], binning_endpoints=None,
         fig['layout']['yaxis']['range'][0] = center[1] - new_height * 0.5
         fig['layout']['yaxis']['range'][1] = center[1] + new_height * 0.5
 
-    return fig
+    return go.Figure(fig)
@@ -2817,7 +2817,7 @@ def test_full_choropleth(self):
                 simplify_county=1
             )
 
-            exp_fig_head = [
+            exp_fig_head = (
                 -88.053375,
                 -88.02916499999999,
                 -88.02432999999999,
@@ -2868,7 +2868,7 @@ def test_full_choropleth(self):
                 -85.142567,
                 -85.113329,
                 -85.10533699999999
-            ]
+            )
 
             self.assertEqual(fig['data'][2]['x'][:50], exp_fig_head)