8 lines
285 B
Python
8 lines
285 B
Python
from image_prediction.image_extractor.extractor import ImageExtractor, ImageMetadataPair
|
|
|
|
|
|
class ImageExtractorMock(ImageExtractor):
|
|
def extract(self, image_container):
|
|
for i, image in enumerate(image_container):
|
|
yield ImageMetadataPair(image, {"image_id": i})
|