From fd6c111f5e2eb33d1106b57b631da1a635217d8a Mon Sep 17 00:00:00 2001 From: Pierre Marcenac Date: Wed, 17 Apr 2024 01:24:39 -0700 Subject: [PATCH] Use epy.lazy_imports in datasets instead of try/except. PiperOrigin-RevId: 625597756 --- tensorflow_datasets/__init__.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tensorflow_datasets/__init__.py b/tensorflow_datasets/__init__.py index 7d6cfcb1d8f..fdaf874b9fc 100644 --- a/tensorflow_datasets/__init__.py +++ b/tensorflow_datasets/__init__.py @@ -40,6 +40,7 @@ _TIMESTAMP_IMPORT_STARTS = time.time() from absl import logging +from etils import epy as _epy import tensorflow_datasets.core.logging as _tfds_logging from tensorflow_datasets.core.logging import call_metadata as _call_metadata @@ -55,7 +56,7 @@ # pytype: disable=import-error # For builds that don't include all dataset builders, we don't want to fail on # import errors of dataset builders. - try: + with _epy.lazy_api_imports(globals()): from tensorflow_datasets import audio from tensorflow_datasets import graphs from tensorflow_datasets import image @@ -79,8 +80,6 @@ from tensorflow_datasets import video from tensorflow_datasets import vision_language - except ImportError: - pass # pytype: enable=import-error _import_time_ms_dataset_builders = int( @@ -103,3 +102,5 @@ import_time_ms_tensorflow=0, import_time_ms_dataset_builders=_import_time_ms_dataset_builders, ) + +del _epy