suppress

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")