Merge pull request #858 from martinthomson/take-n-ba

Cap the iterator that comes out of BitArr
private-attribution · Nov 24, 2023 · 20224b7 · 20224b7
2 parents be5e304 + e23a446
commit 20224b7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ff/boolean_array.rs b/src/ff/boolean_array.rs
@@ -27,7 +27,7 @@ macro_rules! store_impl {
 
 /// iterator for Boolean arrays
 pub struct BAIterator<'a> {
-    iterator: Iter<'a, u8, Lsb0>,
+    iterator: std::iter::Take<Iter<'a, u8, Lsb0>>,
 }
 
 ///impl Iterator for all Boolean arrays
@@ -228,7 +228,7 @@ macro_rules! boolean_array_impl {
 
                 fn into_iter(self) -> Self::IntoIter {
                     BAIterator {
-                        iterator: self.0.iter(),
+                        iterator: self.0.iter().take(usize::try_from(<$name>::BITS).unwrap()),
                     }
                 }
             }