From 739b7fb52acd5a7dc98bc6af57230d12ea4ebf45 Mon Sep 17 00:00:00 2001 From: Doojin Park Date: Wed, 4 Dec 2024 06:35:22 +0000 Subject: [PATCH] update taxi_utils not to use deprecated model save apis --- tfx/examples/chicago_taxi_pipeline/taxi_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tfx/examples/chicago_taxi_pipeline/taxi_utils.py b/tfx/examples/chicago_taxi_pipeline/taxi_utils.py index 2ca7167b22..214aa29de9 100644 --- a/tfx/examples/chicago_taxi_pipeline/taxi_utils.py +++ b/tfx/examples/chicago_taxi_pipeline/taxi_utils.py @@ -370,4 +370,4 @@ def run_fn(fn_args: fn_args_utils.FnArgs): model, tf_transform_output ), } - model.save(fn_args.serving_model_dir, save_format='tf', signatures=signatures) + tf.saved_model.save(model, fn_args.serving_model_dir, signatures=signatures)