Update image_classification.rst

Lightning-Universe · Feb 19, 2022 · f74f99e · f74f99e
1 parent fcabf95
commit f74f99e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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