Skip to content

Commit 1ad0f74

Browse files
authored
Add RandomTranslation to serialization test (#1315)
* Add RandomTranslation to KerasCV Let's keep this PR lightweight for now, then we can tidy the layer to fit standards later. Right now we are broken per Keras' latest release. * Add copyright * Lint fix * Fix mixed precision test * Fix random transltion layers
1 parent a549c35 commit 1ad0f74

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

keras_cv/layers/serialization_test.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,11 @@ class SerializationTest(tf.test.TestCase, parameterized.TestCase):
111111
cv_layers.RandomChannelShift,
112112
{"value_range": (0, 255), "factor": 0.5},
113113
),
114+
(
115+
"RandomTranslation",
116+
cv_layers.RandomTranslation,
117+
{"width_factor": (0, 0.5), "height_factor": 0.5},
118+
),
114119
(
115120
"Posterization",
116121
cv_layers.Posterization,

0 commit comments

Comments
 (0)