Skip to content

Commit 307ebd8

Browse files
committed
Fix to make the test more stable.
Change-Id: Id9edf30a211cdace462737e2f390d6d697ec458c
1 parent 7dc76a4 commit 307ebd8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow_model_optimization/python/core/clustering/keras/cluster_wrapper_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,7 @@ def testValuesAreClusteredAfterStripping(self,
170170
or equal to number_of_clusters.
171171
"""
172172
original_model = tf.keras.Sequential([
173-
layers.Dense(32, input_shape=(10,)),
173+
layers.Dense(32, input_shape=(10,), name='abc'),
174174
])
175175

176176
weights_name = original_model.layers[0].weights[0].name

0 commit comments

Comments
 (0)