better cast to generator

nicholas-esterer · nicholas-esterer · commit 30970328397a · 2020-11-12T11:55:48.000-05:00
diff --git a/packages/python/plotly/plotly/basedatatypes.py b/packages/python/plotly/plotly/basedatatypes.py
@@ -156,8 +156,8 @@ def _axis_spanning_shapes_docstr(shape_type):
 
 def _generator(i):
     """ "cast" an iterator to a generator """
-    while True:
-        yield next(i)
+    for x in i:
+        yield x
 
 
 class BaseFigure(object):
