Add output redirection through InferrerStep setter

0xKoios · Feb 5, 2021 · 9349e71 · 9349e71
1 parent d8b4a9a
commit 9349e71
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/zenml/core/components/bulk_inferrer/executor.py b/zenml/core/components/bulk_inferrer/executor.py
@@ -147,6 +147,7 @@ def _run_model_inference(
             for split, example_uri in example_uris.items():
                 output_examples_split_uri = artifact_utils.get_split_uri(
                     [output_examples], split)
+                inferrer_step.set_output_uri(output_examples_split_uri)
                 logging.info('Path of output examples split `%s` is %s.',
                              split, output_examples_split_uri)
                 _ = (