Pass neuralization progress callback to the backend

LukasZahradnik · Apr 3, 2024 · 477fcc0 · 477fcc0
1 parent 4847d0e
commit 477fcc0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neuralogic/core/builder/builder.py b/neuralogic/core/builder/builder.py
@@ -76,7 +76,7 @@ def neuralize(self, groundings, progress: bool, length: Optional[int]) -> List[N
             return self._neuralize(groundings, self._callback(pbar))
 
     def _neuralize(self, groundings, callback) -> List[NeuralSample]:
-        neuralize_pipeline = self.example_builder.neuralize(groundings, None)
+        neuralize_pipeline = self.example_builder.neuralize(groundings, callback)
         neuralize_pipeline.execute(None)
 
         samples = neuralize_pipeline.get()