Move assignment implementation into own file

scroll-tech · Nov 6, 2023 · f73694b · f73694b
1 parent f1bce41
commit f73694b
Show file tree

Hide file tree

Showing 4 changed files with 674 additions and 537 deletions.
diff --git a/src/assignment_map.rs b/src/assignment_map.rs
@@ -8,7 +8,7 @@ use halo2_proofs::{
 };
 use itertools::Itertools;
 use rayon::prelude::*;
-use std::collections::{BTreeMap, BTreeSet, HashMap};
+use std::collections::BTreeMap;
 
 #[derive(Clone, Default)]
 pub struct AssignmentMap<F: FieldExt>(BTreeMap<usize, Vec<(Column, Value<F>)>>);
@@ -34,10 +34,10 @@ impl<F: FieldExt> AssignmentMap<F> {
         Self(y)
     }
 
-    pub fn to_vec(self) -> Vec<impl FnMut(Region<'_, F>) -> Result<(), Error>> {
+    pub fn into_vec(self) -> Vec<impl FnMut(Region<'_, F>) -> Result<(), Error>> {
         self.0
-            .into_iter()
-            .map(|(_offset, column_assignments)| {
+            .into_values()
+            .map(|column_assignments| {
                 move |mut region: Region<'_, F>| {
                     for (column, value) in column_assignments.iter() {
                         match *column {