Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Commit

Permalink
Fix
Browse files Browse the repository at this point in the history
  • Loading branch information
@ethanwharris
ethanwharris committed Jan 25, 2022
1 parent 1d3cebc commit a21e29c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions flash/image/detection/data.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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.*"]


Expand Down

0 comments on commit a21e29c

Please sign in to comment.