better cast to generator

nicholas-esterer · nicolaskruchten · commit d4302cc8de0f · 2020-11-18T15:24:34.000-05:00
diff --git a/packages/python/plotly/plotly/basedatatypes.py b/packages/python/plotly/plotly/basedatatypes.py
@@ -384,8 +384,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):
