From a21e29c6ec264527d51a55f5a24752931ced36f3 Mon Sep 17 00:00:00 2001 From: Ethan Harris Date: Tue, 25 Jan 2022 18:34:11 +0000 Subject: [PATCH] Fix --- flash/image/detection/data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/flash/image/detection/data.py b/flash/image/detection/data.py index 00f0ffcdc9..93fc1e2a1d 100644 --- a/flash/image/detection/data.py +++ b/flash/image/detection/data.py @@ -20,7 +20,7 @@ from flash.core.data.utilities.sort import sorted_alphanumeric from flash.core.integrations.icevision.data import IceVisionInput from flash.core.integrations.icevision.transforms import IceVisionInputTransform -from flash.core.utilities.imports import _FIFTYONE_AVAILABLE, _ICEVISION_AVAILABLE, requires +from flash.core.utilities.imports import _FIFTYONE_AVAILABLE, _ICEVISION_AVAILABLE, _IMAGE_EXTRAS_TESTING, requires from flash.core.utilities.stages import RunningStage from flash.core.utilities.types import INPUT_TRANSFORM_TYPE from flash.image.detection.input import ObjectDetectionFiftyOneInput @@ -40,7 +40,7 @@ Parser = object # Skip doctests if requirements aren't available -if not _ICEVISION_AVAILABLE: +if not _IMAGE_EXTRAS_TESTING: __doctest_skip__ = ["ObjectDetectionData", "ObjectDetectionData.*"]