diff --git a/keras_cv/layers/regularization/dropblock_2d_test.py b/keras_cv/layers/regularization/dropblock_2d_test.py index 33a1c11e3b..c86e08b3ef 100644 --- a/keras_cv/layers/regularization/dropblock_2d_test.py +++ b/keras_cv/layers/regularization/dropblock_2d_test.py @@ -14,6 +14,7 @@ import pytest import tensorflow as tf + from keras_cv.backend.config import keras_3 from keras_cv.layers.regularization.dropblock_2d import DropBlock2D from keras_cv.tests.test_case import TestCase