src/cloud/pipelines/image_classification/preprocessing.py (2 lines): - line 137: # TODO: Only augment training images, not entire dataset! - line 154: # FIXME: only augment train dataset, not the test dataset! src/cloud/pipelines/image_classification/evaluation.py (1 line): - line 103: # TODO: make batch prediction work