Release source when PyIterReturn::StopIteration

rng-dynamics · rng-dynamics · commit 76ef074b5a13 · 2022-06-14T18:59:27.000+02:00
diff --git a/vm/src/stdlib/itertools.rs b/vm/src/stdlib/itertools.rs
@@ -96,10 +96,13 @@ mod decl {
                     }
                 }
             };
-            next.map_err(|err| {
-                *zelf.source.write() = None;
-                err
-            })
+            match next {
+                Err(_) | Ok(PyIterReturn::StopIteration(_)) => {
+                    *zelf.source.write() = None;
+                }
+                _ => {}
+            };
+            next
         }
     }
 

Original file line number	Diff line number	Diff line change
`@@ -96,10 +96,13 @@ mod decl {`
`96`	`96`	`}`
`97`	`97`	`}`
`98`	`98`	`};`
`99`		`- next.map_err(\|err\| {`
`100`		`- *zelf.source.write() = None;`
`101`		`- err`
`102`		`- })`
	`99`	`+ match next {`
	`100`	`+ Err(_) \| Ok(PyIterReturn::StopIteration(_)) => {`
	`101`	`+ *zelf.source.write() = None;`
	`102`	`+ }`
	`103`	`+ _ => {}`
	`104`	`+ };`
	`105`	`+ next`
`103`	`106`	`}`
`104`	`107`	`}`
`105`	`108`