Skip to content

Commit

Permalink
Internal change
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 683623398
  • Loading branch information
achoum authored and copybara-github committed Oct 8, 2024
1 parent 0b4a3bd commit fcae77f
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions yggdrasil_decision_forests/port/python/ydf/model/export_tf.py
Original file line number Diff line number Diff line change
Expand Up @@ -131,6 +131,17 @@ def ydf_model_to_tensorflow_saved_model(
]:
if value != expected:
raise ValueError(f"{name!r} is not supported for `keras` mode.")

if input_model_signature_fn is None:
logging.warning(
'to_tensorflow_saved_model with mode="keras" is deprecated. Use'
' mode="tf" instead. See:'
" https://ydf.readthedocs.io/en/latest/py_api/GenericModel/#ydf.GenericModel.to_tensorflow_function."
" mode='tf' natively support Servo API and other functionnalities"
" while at the same time creating a simpler and easier to use"
" SavedModel."
)

ydf_model_to_tensorflow_saved_model_keras_mode(
ydf_model=ydf_model,
path=path,
Expand Down

0 comments on commit fcae77f

Please sign in to comment.