diff --git a/image_prediction/pipeline.py b/image_prediction/pipeline.py index 9621ae9..8ac9d7b 100644 --- a/image_prediction/pipeline.py +++ b/image_prediction/pipeline.py @@ -39,9 +39,9 @@ class Pipeline: classify = compose(chain.from_iterable, lift(classifier), partial(chunks, batch_size)) join = compose(starlift(lambda prd, mdt: {"classification": prd, **mdt}), splat(zip)) - # +>--classify--v - # --extract image metadata pairs-->--split--| |--join-->format - # +>--identity--^ + # +>--classify--v + # --extract--| |--join-->format + # +>--identity--^ self.pipe = rcompose( extract, # ... image-metadata-pairs as a stream