Improve error handling for missing TensorFlow dependency in KerasCV

keras_cv/src/backend/tf_ops.py

@@ -46,7 +46,14 @@
 from tensorflow import split  # noqa: F403, F401
 
 import numpy as np
-import tensorflow as tf
+
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 
 def smart_resize(x, size, interpolation="bilinear"):

keras_cv/src/bounding_box/converters.py

@@ -14,7 +14,13 @@
 """Converter functions for working with bounding box formats."""
 
 
-import tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 from keras_cv.src.api_export import keras_cv_export
 from keras_cv.src.backend import keras

keras_cv/src/bounding_box/converters_test.py

@@ -16,7 +16,14 @@
 
 import numpy as np
 import pytest
-import tensorflow as tf
+
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 from absl.testing import parameterized
 
 from keras_cv.src import bounding_box

keras_cv/src/bounding_box/mask_invalid_detections_test.py

@@ -14,7 +14,14 @@
 
 import numpy as np
 import pytest
-import tensorflow as tf
+
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 from keras_cv.src import bounding_box
 from keras_cv.src.backend import ops

keras_cv/src/bounding_box/to_dense.py

@@ -11,7 +11,13 @@
 # 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 tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 import keras_cv.src.bounding_box.validate_format as validate_format
 from keras_cv.src.api_export import keras_cv_export

keras_cv/src/bounding_box/to_dense_test.py

@@ -12,7 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import pytest
-import tensorflow as tf
+
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 from keras_cv.src import bounding_box
 from keras_cv.src.tests.test_case import TestCase

keras_cv/src/bounding_box/to_ragged.py

@@ -11,7 +11,13 @@
 # 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 tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 import keras_cv.src.bounding_box.validate_format as validate_format
 from keras_cv.src import backend

keras_cv/src/bounding_box/validate_format.py

@@ -11,7 +11,13 @@
 # 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 tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 from keras_cv.src.api_export import keras_cv_export
 

keras_cv/src/callbacks/waymo_evaluation_callback.py

@@ -11,7 +11,13 @@
 # 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 tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 from tensorflow.keras.callbacks import Callback
 
 from keras_cv.src.api_export import keras_cv_export

keras_cv/src/callbacks/waymo_evaluation_callback_test.py

@@ -14,7 +14,14 @@
 
 import numpy as np
 import pytest
-import tensorflow as tf
+
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 from tensorflow import keras
 
 from keras_cv.src.callbacks import WaymoEvaluationCallback

keras_cv/src/core/factor_sampler/constant_factor_sampler.py

@@ -12,7 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 from keras_cv.src.api_export import keras_cv_export
 from keras_cv.src.core.factor_sampler.factor_sampler import FactorSampler

keras_cv/src/core/factor_sampler/normal_factor_sampler.py

@@ -12,7 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 from keras_cv.src.api_export import keras_cv_export
 from keras_cv.src.core.factor_sampler.factor_sampler import FactorSampler

keras_cv/src/core/factor_sampler/uniform_factor_sampler.py

@@ -12,7 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 from keras_cv.src.api_export import keras_cv_export
 from keras_cv.src.core.factor_sampler.factor_sampler import FactorSampler

keras_cv/src/datasets/imagenet/load.py

@@ -12,7 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 from tensorflow.keras import layers
 
 from keras_cv.src.api_export import keras_cv_export

keras_cv/src/datasets/pascal_voc/load.py

@@ -12,7 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 import tensorflow_datasets as tfds
 
 from keras_cv.src import bounding_box

keras_cv/src/datasets/pascal_voc/segmentation.py

@@ -43,7 +43,14 @@ class and instance segmentation masks.
 import xml
 
 import numpy as np
-import tensorflow as tf
+
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 import tensorflow_datasets as tfds
 from tensorflow import keras
 

keras_cv/src/datasets/pascal_voc/segmentation_test.py

@@ -15,7 +15,13 @@
 import pathlib
 import sys
 
-import tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 from absl import flags
 
 from keras_cv.src.datasets.pascal_voc import segmentation

keras_cv/src/datasets/waymo/load.py

@@ -14,7 +14,13 @@
 """Data loader for the Waymo Open Dataset."""
 import os
 
-import tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 from keras_cv.src.datasets.waymo import transformer
 from keras_cv.src.utils import assert_waymo_open_dataset_installed

keras_cv/src/datasets/waymo/struct.py

@@ -16,7 +16,13 @@
 import dataclasses
 from typing import Optional
 
-import tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 
 @dataclasses.dataclass

keras_cv/src/datasets/waymo/transformer.py

@@ -20,7 +20,14 @@
 from typing import Tuple
 
 import numpy as np
-import tensorflow as tf
+
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 from keras_cv.src.api_export import keras_cv_export
 from keras_cv.src.utils import assert_waymo_open_dataset_installed

keras_cv/src/datasets/waymo/transformer_test.py

@@ -15,7 +15,14 @@
 
 import numpy as np
 import pytest
-import tensorflow as tf
+
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 from keras_cv.src.tests.test_case import TestCase
 

keras_cv/src/keypoint/converters.py

@@ -13,7 +13,13 @@
 # limitations under the License.
 """Converter functions for working with keypoints formats."""
 
-import tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 from keras_cv.src.api_export import keras_cv_export
 

keras_cv/src/keypoint/converters_test.py

@@ -15,7 +15,14 @@
 import itertools
 
 import numpy as np
-import tensorflow as tf
+
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 from absl.testing import parameterized
 
 from keras_cv.src import keypoint

keras_cv/src/keypoint/utils.py

@@ -12,7 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Utility functions for keypoint transformation."""
-import tensorflow as tf
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 
 from keras_cv.src.api_export import keras_cv_export
 

keras_cv/src/keypoint/utils_test.py

@@ -13,7 +13,14 @@
 # limitations under the License.
 
 import numpy as np
-import tensorflow as tf
+
+try:
+    import tensorflow as tf
+except ImportError:
+    raise ImportError(
+        "To use KerasCV, please install TensorFlow: `pip install tensorflow`. "
+        "The TensorFlow package is required for data preprocessing with any backend."
+    )
 from absl.testing import parameterized
 
 from keras_cv.src.keypoint.utils import filter_out_of_image