From f74f99e6117b617a716d5e6e3c8dc961f1357dbe Mon Sep 17 00:00:00 2001 From: Ethan Harris Date: Sat, 19 Feb 2022 12:52:46 +0000 Subject: [PATCH] Update image_classification.rst --- docs/source/reference/image_classification.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/reference/image_classification.rst b/docs/source/reference/image_classification.rst index 1ea26b0119..6906cf2b8f 100644 --- a/docs/source/reference/image_classification.rst +++ b/docs/source/reference/image_classification.rst @@ -101,7 +101,7 @@ Here's an example: from torchvision import transforms as T - from typing import Tuple, Callable + from typing import Callable, Tuple, Union import flash from flash.image import ImageClassificationData, ImageClassifier from flash.core.data.io.input_transform import InputTransform