correct errors#

keras-team · Nov 15, 2023 · 275c701 · 275c701
1 parent 6758d69
commit 275c701
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/keras_cv/layers/preprocessing/random_translation.py b/keras_cv/layers/preprocessing/random_translation.py
@@ -15,8 +15,8 @@
 import tensorflow as tf
 
 from keras_cv import bounding_box
-from keras_cv import random
 from keras_cv.api_export import keras_cv_export
+from keras_cv.backend import random
 from keras_cv.layers.preprocessing.vectorized_base_image_augmentation_layer import (  # noqa: E501
     VectorizedBaseImageAugmentationLayer,
 )

diff --git a/keras_cv/layers/preprocessing_3d/base_augmentation_layer_3d.py b/keras_cv/layers/preprocessing_3d/base_augmentation_layer_3d.py
@@ -16,6 +16,7 @@
 
 from keras_cv.api_export import keras_cv_export
 from keras_cv.backend import config
+from keras_cv.backend.random import SeedGenerator
 
 if config.keras_3():
     base_layer = tf.keras.layers.Layer