fix: correct bug in trivial search

antecedent/image.py

@@ -8,6 +8,7 @@ import numpy as np
 
 from antecedent.block import Block
 from antecedent.pipeline import NaivePipeline, create_pipeline
+from antecedent.utils import round
 
 
 class Image:
@@ -152,7 +153,7 @@ class Image:
         if pipeline is None:
             pipeline = self.pipeline
         block_values = np.stack([block.value.reshape(-1, 8, 8) for block in blocks])
-        starts = pipeline.backward(block_values)
+        starts = round(pipeline.backward(block_values))
         transformed_starts = pipeline.forward(starts)
         completed = 0
         for i in range(len(starts)):