We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5a18f53 commit f8e874fCopy full SHA for f8e874f
tensorflow_model_optimization/python/core/common/keras/compression/algorithms/weight_clustering_test.py
@@ -16,6 +16,7 @@
16
17
import os
18
import tempfile
19
+import unittest
20
21
import tensorflow as tf
22
@@ -87,6 +88,8 @@ def _get_directory_size_in_bytes(directory):
87
88
89
class FunctionalTest(tf.test.TestCase):
90
91
+ # TODO(b/246652360): Re-Enable the test once it is fixed.
92
+ @unittest.skip('Test needs to be fixed')
93
def testWeightClustering_TrainingE2E(self):
94
number_of_clusters = 8
95
model = _build_model()
0 commit comments