Skip to content

Commit 8e88992

Browse files
committed
Rename serving input fn
1 parent 036aac7 commit 8e88992

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

00_Miscellaneous/tfx/02_tfx_end_to_end.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1091,7 +1091,7 @@
10911091
"source": [
10921092
"tf.logging.set_verbosity(tf.logging.ERROR)\n",
10931093
"\n",
1094-
"def make_serving_input_receiver_fn():\n",
1094+
"def serving_input_receiver_fn():\n",
10951095
" \n",
10961096
" source_raw_schema = tfdv.load_schema_text(RAW_SCHEMA_LOCATION)\n",
10971097
" raw_feature_spec = schema_utils.schema_as_feature_spec(source_raw_schema).feature_spec\n",
@@ -1137,7 +1137,7 @@
11371137
" \n",
11381138
"estimator.export_savedmodel(\n",
11391139
" export_dir_base=export_dir,\n",
1140-
" serving_input_receiver_fn=make_serving_input_receiver_fn\n",
1140+
" serving_input_receiver_fn=input_receiver_fn\n",
11411141
")"
11421142
]
11431143
},

0 commit comments

Comments
 (0)