Skip to content

Commit

Permalink
update taxi_utils not to use deprecated model save apis
Browse files Browse the repository at this point in the history
  • Loading branch information
nikelite committed Dec 4, 2024
1 parent 14d9822 commit 739b7fb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tfx/examples/chicago_taxi_pipeline/taxi_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

0 comments on commit 739b7fb

Please sign in to comment.