diff --git a/src/flash/core/utilities/imports.py b/src/flash/core/utilities/imports.py index cb118ddf86..656f27f42c 100644 --- a/src/flash/core/utilities/imports.py +++ b/src/flash/core/utilities/imports.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. +import contextlib import functools import importlib import operator @@ -19,6 +20,7 @@ from lightning_utilities.core.imports import compare_version, module_available from pkg_resources import DistributionNotFound +from typeshed_client.typeshed.contextlib import suppress try: from packaging.version import Version @@ -41,7 +43,8 @@ _MATPLOTLIB_AVAILABLE = module_available("matplotlib") _TRANSFORMERS_AVAILABLE = module_available("transformers") _PYSTICHE_AVAILABLE = module_available("pystiche") -_FIFTYONE_AVAILABLE = module_available("fiftyone") +with contextlib.suppress(ConnectionError): + _FIFTYONE_AVAILABLE = module_available("fiftyone") _FASTAPI_AVAILABLE = module_available("fastapi") _PYDANTIC_AVAILABLE = module_available("pydantic") _GRAPHVIZ_AVAILABLE = module_available("graphviz")