Skip to content

Commit 480f5f1

Browse files
committed
fix tests
1 parent 76ef074 commit 480f5f1

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

vm/src/stdlib/itertools.rs

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -69,8 +69,8 @@ mod decl {
6969
return Ok(PyIterReturn::StopIteration(None));
7070
};
7171
let next = loop {
72-
let option_active = zelf.active.read().clone();
73-
if let Some(active) = option_active {
72+
let maybe_active = zelf.active.read().clone();
73+
if let Some(active) = maybe_active {
7474
match active.next(vm) {
7575
Ok(PyIterReturn::Return(ok)) => {
7676
break Ok(PyIterReturn::Return(ok));
@@ -84,9 +84,14 @@ mod decl {
8484
}
8585
} else {
8686
match source.next(vm) {
87-
Ok(PyIterReturn::Return(ok)) => {
88-
*zelf.active.write() = Some(ok.get_iter(vm)?);
89-
}
87+
Ok(PyIterReturn::Return(ok)) => match ok.get_iter(vm) {
88+
Ok(iter) => {
89+
*zelf.active.write() = Some(iter);
90+
}
91+
Err(err) => {
92+
break Err(err);
93+
}
94+
},
9095
Ok(PyIterReturn::StopIteration(_)) => {
9196
break Ok(PyIterReturn::StopIteration(None));
9297
}

0 commit comments

Comments
 (0)