fix(api): report chain progress using a new callback if none was provided

2023-09-12 18:19:07 -05:00 · 2023-09-12 18:19:07 -05:00 · f9acf9b50f
parent 55ddb9fdac
commit f9acf9b50f
1 changed files with 3 additions and 1 deletions
--- a/api/onnx_web/chain/base.py
+++ b/api/onnx_web/chain/base.py
@ -105,7 +105,9 @@ class ChainPipeline:
        """
        DEPRECATED: use `run` instead
        """
-        if callback is not None:
+        if callback is None:
+            callback = worker.get_progress_callback()
+        else:
            callback = ChainProgress.from_progress(callback)

        start = monotonic()