Skip to content

Commit 8e790c6

Browse files
committed
Addressed reviewer's comments.
Change-Id: I05c69e81f7635a1565bc4f8ece09c1ec736b7878
1 parent c5f42ab commit 8e790c6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -120,6 +120,7 @@ def testMnist(self):
120120

121121
# Record the number of unique values of 'bias'
122122
nr_of_bias_weights = _get_number_of_unique_weights(model, -1, 1)
123+
self.assertGreater(nr_of_bias_weights, NUMBER_OF_CLUSTERS)
123124

124125
_, (x_test, y_test) = _get_dataset()
125126

0 commit comments

Comments
 (0)