Skip to content

Commit b68b4da

Browse files
nkovela1tfx-copybara
authored andcommitted
Ensures serialization in legacy format for TFMA metric wrapper.
PiperOrigin-RevId: 518128574
1 parent 958a0e7 commit b68b4da

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tensorflow_model_analysis/metrics/tf_metric_wrapper.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -267,10 +267,12 @@ def _metric_keys_and_configs(
267267
example_weighted=example_weighted))
268268
if isinstance(metric, tf.keras.metrics.Metric):
269269
metric_configs[updated_sub_key][output_name].append(
270-
metric_util.serialize_metric(metric))
270+
metric_util.serialize_metric(metric, use_legacy_format=True)
271+
)
271272
elif isinstance(metric, tf.keras.losses.Loss):
272273
loss_configs[updated_sub_key][output_name].append(
273-
metric_util.serialize_loss(metric))
274+
metric_util.serialize_loss(metric, use_legacy_format=True)
275+
)
274276
return metric_keys, metric_configs, loss_configs
275277

276278

0 commit comments

Comments
 (0)