Allow to use teh iterator for results

1 files changed, 3 insertions, 3 deletions

diff --git a/src/solvers/mod.rs b/src/solvers/mod.rs
index 189deca..5b7a495 100644
--- a/src/solvers/mod.rs
+++ b/src/solvers/mod.rs
@@ -3,7 +3,7 @@ pub mod gpu;
 pub mod gpusolver;
 pub mod single;
 
-use crate::structs::StoneWall;
+//use crate::structs::StoneWall;
 pub use gpu::*;
 
 /// calculate h and w
@@ -20,10 +20,10 @@ pub trait Solver {
 }
 
 pub trait FirstSolver {
-    fn solve(self) -> StoneWall;
+    fn solve(self) -> RowResult;
 }
 
 pub trait IteratorSolver: Solver {
-    type IntoIter: Iterator<Item = StoneWall>;
+    type IntoIter: Iterator<Item = RowResult>;
     fn solve(self) -> Self::IntoIter;
 }