diff --git a/image_prediction/extractor_classifier/extractor_classifier.py b/image_prediction/extractor_classifier/extractor_classifier.py index 83f2b2c..eef74da 100644 --- a/image_prediction/extractor_classifier/extractor_classifier.py +++ b/image_prediction/extractor_classifier/extractor_classifier.py @@ -10,7 +10,8 @@ class ExtractorClassifier: """This class is responsible for orchestrating the pairing of classifications and image metadata. It extracts images from an object and classifies them. Then it ties the classification together with the metadata. It returns an iterable of dictionaries, where each dictionary has a field 'label' for the classification and possibly additional - fields for metadata -- metadata could be void.""" + fields for metadata -- metadata could be void. + """ def __init__(self, image_extractor: ImageExtractor, image_classifier: ImageClassifier): self.classifier = image_classifier