Merge branch 'main' into setup

zmrocze · Jul 25, 2024 · c678d41 · c678d41
2 parents 71de08e + 5e5c663
commit c678d41
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/lib/src/subcommands/setup.rs b/lib/src/subcommands/setup.rs
@@ -1,11 +1,13 @@
 use std::{
+  collections::HashMap,
   fs::File,
   path::{Path, PathBuf},
 };
 
 use ark_groth16::Groth16;
 use ark_serialize::CanonicalSerialize;
 use ark_snark::SNARK;
+use luminal::compiler_utils::ToId;
 use rand::{rngs::StdRng, SeedableRng};
 
 use crate::{model::TrainParams, snark::Curve};
@@ -33,7 +35,7 @@ impl Setup {
     }
   }
 
-  pub async fn run(self) {
+  pub fn run(self) {
     let rng = StdRng::seed_from_u64(1);
 
     let dataset = crate::model::read_dataset(self.dataset_path.as_path()).unwrap();

diff --git a/lib/src/utils.rs b/lib/src/utils.rs
@@ -1,3 +1,7 @@
+use ark_serialize::{CanonicalSerialize, Write};
+use serde::Serialize;
+use std::path::Path;
+
 #[cfg(not(debug_assertions))]
 use human_panic::setup_panic;