File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
tensorflow_model_optimization/python/core/clustering/keras Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -133,7 +133,7 @@ def _add_clustering_wrapper(layer):
133
133
# There is no available public API to do this check.
134
134
if (not layer ._is_graph_network and
135
135
not isinstance (layer , keras .models .Sequential )):
136
- raise ValueError ("SubClass models are not supported currently." )
136
+ raise ValueError ("Subclassed models are not supported currently." )
137
137
138
138
return keras .models .clone_model (layer ,
139
139
input_tensors = None ,
Original file line number Diff line number Diff line change @@ -403,7 +403,7 @@ def testClusterSubclassModelAsSubmodel(self):
403
403
layers .Dense (10 ),
404
404
model_subclass
405
405
])
406
- with self .assertRaisesRegexp (ValueError , "SubClass models.*" ):
406
+ with self .assertRaisesRegexp (ValueError , "Subclassed models.*" ):
407
407
_ = cluster .cluster_weights (model , ** self .params )
408
408
409
409
@keras_parameterized .run_all_keras_modes
You can’t perform that action at this time.
0 commit comments