fix import

keras-team · Nov 18, 2023 · 29f35eb · 29f35eb
1 parent fb6c471
commit 29f35eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_cv/layers/preprocessing/vectorized_base_image_augmentation_layer_test.py b/keras_cv/layers/preprocessing/vectorized_base_image_augmentation_layer_test.py
@@ -18,7 +18,7 @@
 from keras_cv.backend import ops
 from keras_cv.layers.preprocessing.vectorized_base_image_augmentation_layer import \
     VectorizedBaseImageAugmentationLayer  # noqa: E501
-
+from keras_cv.tests.test_case import TestCase
 
 
 class VectorizedRandomAddLayer(VectorizedBaseImageAugmentationLayer):