Update version for release 0.4.1

PiperOrigin-RevId: 386126448

tensorflow_ranking/g3doc/api_docs/python/tfr.md

@@ -56,7 +56,7 @@ library.
 __version__<a id="__version__"></a>
 </td>
 <td>
-`'0.4.0.dev'`
+`'0.4.1.dev'`
 </td>
 </tr>
 </table>

tensorflow_ranking/g3doc/api_docs/python/tfr/_api_cache.json

@@ -1263,6 +1263,48 @@
     "tfr.keras.strategy_utils.NullContextManager.__lt__": "tfr.keras.canned.DNNRankingNetwork.__lt__",
     "tfr.keras.strategy_utils.NullContextManager.__ne__": "tfr.keras.canned.DNNRankingNetwork.__ne__",
     "tfr.keras.strategy_utils.NullContextManager.__new__": "tfr.keras.losses.ApproxMRRLoss.__new__",
+    "tfr.keras.task.RankingDataConfig.__ge__": "tfr.keras.canned.DNNRankingNetwork.__ge__",
+    "tfr.keras.task.RankingDataConfig.__gt__": "tfr.keras.canned.DNNRankingNetwork.__gt__",
+    "tfr.keras.task.RankingDataConfig.__le__": "tfr.keras.canned.DNNRankingNetwork.__le__",
+    "tfr.keras.task.RankingDataConfig.__lt__": "tfr.keras.canned.DNNRankingNetwork.__lt__",
+    "tfr.keras.task.RankingDataConfig.__ne__": "tfr.keras.canned.DNNRankingNetwork.__ne__",
+    "tfr.keras.task.RankingDataConfig.__new__": "tfr.keras.losses.ApproxMRRLoss.__new__",
+    "tfr.keras.task.RankingDataLoader.__eq__": "tfr.keras.canned.DNNRankingNetwork.__eq__",
+    "tfr.keras.task.RankingDataLoader.__ge__": "tfr.keras.canned.DNNRankingNetwork.__ge__",
+    "tfr.keras.task.RankingDataLoader.__gt__": "tfr.keras.canned.DNNRankingNetwork.__gt__",
+    "tfr.keras.task.RankingDataLoader.__le__": "tfr.keras.canned.DNNRankingNetwork.__le__",
+    "tfr.keras.task.RankingDataLoader.__lt__": "tfr.keras.canned.DNNRankingNetwork.__lt__",
+    "tfr.keras.task.RankingDataLoader.__ne__": "tfr.keras.canned.DNNRankingNetwork.__ne__",
+    "tfr.keras.task.RankingDataLoader.__new__": "tfr.keras.losses.ApproxMRRLoss.__new__",
+    "tfr.keras.task.RankingTask.__eq__": "tfr.keras.canned.DNNRankingNetwork.__eq__",
+    "tfr.keras.task.RankingTask.__ge__": "tfr.keras.canned.DNNRankingNetwork.__ge__",
+    "tfr.keras.task.RankingTask.__gt__": "tfr.keras.canned.DNNRankingNetwork.__gt__",
+    "tfr.keras.task.RankingTask.__le__": "tfr.keras.canned.DNNRankingNetwork.__le__",
+    "tfr.keras.task.RankingTask.__lt__": "tfr.keras.canned.DNNRankingNetwork.__lt__",
+    "tfr.keras.task.RankingTask.__ne__": "tfr.keras.canned.DNNRankingNetwork.__ne__",
+    "tfr.keras.task.RankingTask.__new__": "tfr.keras.losses.ApproxMRRLoss.__new__",
+    "tfr.keras.task.RankingTask.name": "tfr.keras.saved_model.Signatures.name",
+    "tfr.keras.task.RankingTask.name_scope": "tfr.keras.canned.DNNRankingNetwork.name_scope",
+    "tfr.keras.task.RankingTask.non_trainable_variables": "tfr.keras.saved_model.Signatures.non_trainable_variables",
+    "tfr.keras.task.RankingTask.submodules": "tfr.keras.canned.DNNRankingNetwork.submodules",
+    "tfr.keras.task.RankingTask.trainable_variables": "tfr.keras.saved_model.Signatures.trainable_variables",
+    "tfr.keras.task.RankingTask.variables": "tfr.keras.saved_model.Signatures.variables",
+    "tfr.keras.task.RankingTaskConfig.IMMUTABLE_TYPES": "tfr.keras.task.RankingDataConfig.IMMUTABLE_TYPES",
+    "tfr.keras.task.RankingTaskConfig.RESERVED_ATTR": "tfr.keras.task.RankingDataConfig.RESERVED_ATTR",
+    "tfr.keras.task.RankingTaskConfig.SEQUENCE_TYPES": "tfr.keras.task.RankingDataConfig.SEQUENCE_TYPES",
+    "tfr.keras.task.RankingTaskConfig.__contains__": "tfr.keras.task.RankingDataConfig.__contains__",
+    "tfr.keras.task.RankingTaskConfig.__ge__": "tfr.keras.canned.DNNRankingNetwork.__ge__",
+    "tfr.keras.task.RankingTaskConfig.__gt__": "tfr.keras.canned.DNNRankingNetwork.__gt__",
+    "tfr.keras.task.RankingTaskConfig.__le__": "tfr.keras.canned.DNNRankingNetwork.__le__",
+    "tfr.keras.task.RankingTaskConfig.__lt__": "tfr.keras.canned.DNNRankingNetwork.__lt__",
+    "tfr.keras.task.RankingTaskConfig.__ne__": "tfr.keras.canned.DNNRankingNetwork.__ne__",
+    "tfr.keras.task.RankingTaskConfig.__new__": "tfr.keras.losses.ApproxMRRLoss.__new__",
+    "tfr.keras.task.RankingTaskConfig.as_dict": "tfr.keras.task.RankingDataConfig.as_dict",
+    "tfr.keras.task.RankingTaskConfig.get": "tfr.keras.task.RankingDataConfig.get",
+    "tfr.keras.task.RankingTaskConfig.lock": "tfr.keras.task.RankingDataConfig.lock",
+    "tfr.keras.task.RankingTaskConfig.override": "tfr.keras.task.RankingDataConfig.override",
+    "tfr.keras.task.RankingTaskConfig.replace": "tfr.keras.task.RankingDataConfig.replace",
+    "tfr.keras.task.RankingTaskConfig.validate": "tfr.keras.task.RankingDataConfig.validate",
     "tfr.losses.RankingLossKey.__eq__": "tfr.keras.canned.DNNRankingNetwork.__eq__",
     "tfr.losses.RankingLossKey.__ge__": "tfr.keras.canned.DNNRankingNetwork.__ge__",
     "tfr.losses.RankingLossKey.__gt__": "tfr.keras.canned.DNNRankingNetwork.__gt__",
@@ -2681,6 +2723,7 @@
     "tfr.keras.model.Scorer.__lt__": true,
     "tfr.keras.model.Scorer.__ne__": true,
     "tfr.keras.model.Scorer.__new__": true,
+    "tfr.keras.model.TensorDict": false,
     "tfr.keras.model.TensorLike": false,
     "tfr.keras.model.TypeSpecInputCreator": false,
     "tfr.keras.model.TypeSpecInputCreator.__call__": true,
@@ -2881,11 +2924,15 @@
     "tfr.keras.pipeline.DatasetHparams.dataset_reader.range": true,
     "tfr.keras.pipeline.DatasetHparams.dataset_reader.reduce": true,
     "tfr.keras.pipeline.DatasetHparams.dataset_reader.repeat": true,
+    "tfr.keras.pipeline.DatasetHparams.dataset_reader.scan": true,
     "tfr.keras.pipeline.DatasetHparams.dataset_reader.shard": true,
     "tfr.keras.pipeline.DatasetHparams.dataset_reader.shuffle": true,
     "tfr.keras.pipeline.DatasetHparams.dataset_reader.skip": true,
+    "tfr.keras.pipeline.DatasetHparams.dataset_reader.snapshot": true,
     "tfr.keras.pipeline.DatasetHparams.dataset_reader.take": true,
+    "tfr.keras.pipeline.DatasetHparams.dataset_reader.take_while": true,
     "tfr.keras.pipeline.DatasetHparams.dataset_reader.unbatch": true,
+    "tfr.keras.pipeline.DatasetHparams.dataset_reader.unique": true,
     "tfr.keras.pipeline.DatasetHparams.dataset_reader.window": true,
     "tfr.keras.pipeline.DatasetHparams.dataset_reader.with_options": true,
     "tfr.keras.pipeline.DatasetHparams.dataset_reader.zip": true,
@@ -3032,6 +3079,133 @@
     "tfr.keras.strategy_utils.get_output_filepath": false,
     "tfr.keras.strategy_utils.get_strategy": false,
     "tfr.keras.strategy_utils.strategy_scope": false,
+    "tfr.keras.task": false,
+    "tfr.keras.task.DATASET_FN_MAP": true,
+    "tfr.keras.task.FeatureSpec": false,
+    "tfr.keras.task.MASK": true,
+    "tfr.keras.task.RankingDataConfig": false,
+    "tfr.keras.task.RankingDataConfig.IMMUTABLE_TYPES": true,
+    "tfr.keras.task.RankingDataConfig.RESERVED_ATTR": true,
+    "tfr.keras.task.RankingDataConfig.SEQUENCE_TYPES": true,
+    "tfr.keras.task.RankingDataConfig.__contains__": true,
+    "tfr.keras.task.RankingDataConfig.__eq__": true,
+    "tfr.keras.task.RankingDataConfig.__ge__": true,
+    "tfr.keras.task.RankingDataConfig.__gt__": true,
+    "tfr.keras.task.RankingDataConfig.__init__": true,
+    "tfr.keras.task.RankingDataConfig.__le__": true,
+    "tfr.keras.task.RankingDataConfig.__lt__": true,
+    "tfr.keras.task.RankingDataConfig.__ne__": true,
+    "tfr.keras.task.RankingDataConfig.__new__": true,
+    "tfr.keras.task.RankingDataConfig.as_dict": true,
+    "tfr.keras.task.RankingDataConfig.block_length": true,
+    "tfr.keras.task.RankingDataConfig.cache": true,
+    "tfr.keras.task.RankingDataConfig.convert_labels_to_binary": true,
+    "tfr.keras.task.RankingDataConfig.cycle_length": true,
+    "tfr.keras.task.RankingDataConfig.data_format": true,
+    "tfr.keras.task.RankingDataConfig.dataset_fn": true,
+    "tfr.keras.task.RankingDataConfig.default_params": true,
+    "tfr.keras.task.RankingDataConfig.deterministic": true,
+    "tfr.keras.task.RankingDataConfig.drop_remainder": true,
+    "tfr.keras.task.RankingDataConfig.enable_tf_data_service": true,
+    "tfr.keras.task.RankingDataConfig.from_args": true,
+    "tfr.keras.task.RankingDataConfig.from_json": true,
+    "tfr.keras.task.RankingDataConfig.from_yaml": true,
+    "tfr.keras.task.RankingDataConfig.get": true,
+    "tfr.keras.task.RankingDataConfig.global_batch_size": true,
+    "tfr.keras.task.RankingDataConfig.input_path": true,
+    "tfr.keras.task.RankingDataConfig.is_training": true,
+    "tfr.keras.task.RankingDataConfig.list_size": true,
+    "tfr.keras.task.RankingDataConfig.lock": true,
+    "tfr.keras.task.RankingDataConfig.mask_feature_name": true,
+    "tfr.keras.task.RankingDataConfig.override": true,
+    "tfr.keras.task.RankingDataConfig.replace": true,
+    "tfr.keras.task.RankingDataConfig.restrictions": true,
+    "tfr.keras.task.RankingDataConfig.seed": true,
+    "tfr.keras.task.RankingDataConfig.sharding": true,
+    "tfr.keras.task.RankingDataConfig.shuffle_buffer_size": true,
+    "tfr.keras.task.RankingDataConfig.shuffle_examples": true,
+    "tfr.keras.task.RankingDataConfig.tf_data_service_address": true,
+    "tfr.keras.task.RankingDataConfig.tf_data_service_job_name": true,
+    "tfr.keras.task.RankingDataConfig.tfds_as_supervised": true,
+    "tfr.keras.task.RankingDataConfig.tfds_data_dir": true,
+    "tfr.keras.task.RankingDataConfig.tfds_name": true,
+    "tfr.keras.task.RankingDataConfig.tfds_skip_decoding_feature": true,
+    "tfr.keras.task.RankingDataConfig.tfds_split": true,
+    "tfr.keras.task.RankingDataConfig.validate": true,
+    "tfr.keras.task.RankingDataLoader": false,
+    "tfr.keras.task.RankingDataLoader.__eq__": true,
+    "tfr.keras.task.RankingDataLoader.__ge__": true,
+    "tfr.keras.task.RankingDataLoader.__gt__": true,
+    "tfr.keras.task.RankingDataLoader.__init__": true,
+    "tfr.keras.task.RankingDataLoader.__le__": true,
+    "tfr.keras.task.RankingDataLoader.__lt__": true,
+    "tfr.keras.task.RankingDataLoader.__ne__": true,
+    "tfr.keras.task.RankingDataLoader.__new__": true,
+    "tfr.keras.task.RankingDataLoader.load": true,
+    "tfr.keras.task.RankingTask": false,
+    "tfr.keras.task.RankingTask.__eq__": true,
+    "tfr.keras.task.RankingTask.__ge__": true,
+    "tfr.keras.task.RankingTask.__gt__": true,
+    "tfr.keras.task.RankingTask.__init__": true,
+    "tfr.keras.task.RankingTask.__le__": true,
+    "tfr.keras.task.RankingTask.__lt__": true,
+    "tfr.keras.task.RankingTask.__ne__": true,
+    "tfr.keras.task.RankingTask.__new__": true,
+    "tfr.keras.task.RankingTask.aggregate_logs": true,
+    "tfr.keras.task.RankingTask.build_inputs": true,
+    "tfr.keras.task.RankingTask.build_losses": true,
+    "tfr.keras.task.RankingTask.build_metrics": true,
+    "tfr.keras.task.RankingTask.build_model": true,
+    "tfr.keras.task.RankingTask.create_optimizer": true,
+    "tfr.keras.task.RankingTask.inference_step": true,
+    "tfr.keras.task.RankingTask.initialize": true,
+    "tfr.keras.task.RankingTask.logging_dir": true,
+    "tfr.keras.task.RankingTask.loss": true,
+    "tfr.keras.task.RankingTask.name": true,
+    "tfr.keras.task.RankingTask.name_scope": true,
+    "tfr.keras.task.RankingTask.non_trainable_variables": true,
+    "tfr.keras.task.RankingTask.process_compiled_metrics": true,
+    "tfr.keras.task.RankingTask.process_metrics": true,
+    "tfr.keras.task.RankingTask.reduce_aggregated_logs": true,
+    "tfr.keras.task.RankingTask.submodules": true,
+    "tfr.keras.task.RankingTask.task_config": true,
+    "tfr.keras.task.RankingTask.train_step": true,
+    "tfr.keras.task.RankingTask.trainable_variables": true,
+    "tfr.keras.task.RankingTask.validation_step": true,
+    "tfr.keras.task.RankingTask.variables": true,
+    "tfr.keras.task.RankingTask.with_name_scope": true,
+    "tfr.keras.task.RankingTaskConfig": false,
+    "tfr.keras.task.RankingTaskConfig.IMMUTABLE_TYPES": true,
+    "tfr.keras.task.RankingTaskConfig.RESERVED_ATTR": true,
+    "tfr.keras.task.RankingTaskConfig.SEQUENCE_TYPES": true,
+    "tfr.keras.task.RankingTaskConfig.__contains__": true,
+    "tfr.keras.task.RankingTaskConfig.__eq__": true,
+    "tfr.keras.task.RankingTaskConfig.__ge__": true,
+    "tfr.keras.task.RankingTaskConfig.__gt__": true,
+    "tfr.keras.task.RankingTaskConfig.__init__": true,
+    "tfr.keras.task.RankingTaskConfig.__le__": true,
+    "tfr.keras.task.RankingTaskConfig.__lt__": true,
+    "tfr.keras.task.RankingTaskConfig.__ne__": true,
+    "tfr.keras.task.RankingTaskConfig.__new__": true,
+    "tfr.keras.task.RankingTaskConfig.aggregated_metrics": true,
+    "tfr.keras.task.RankingTaskConfig.as_dict": true,
+    "tfr.keras.task.RankingTaskConfig.default_params": true,
+    "tfr.keras.task.RankingTaskConfig.from_args": true,
+    "tfr.keras.task.RankingTaskConfig.from_json": true,
+    "tfr.keras.task.RankingTaskConfig.from_yaml": true,
+    "tfr.keras.task.RankingTaskConfig.get": true,
+    "tfr.keras.task.RankingTaskConfig.init_checkpoint": true,
+    "tfr.keras.task.RankingTaskConfig.lock": true,
+    "tfr.keras.task.RankingTaskConfig.loss": true,
+    "tfr.keras.task.RankingTaskConfig.loss_reduction": true,
+    "tfr.keras.task.RankingTaskConfig.model": true,
+    "tfr.keras.task.RankingTaskConfig.output_preds": true,
+    "tfr.keras.task.RankingTaskConfig.override": true,
+    "tfr.keras.task.RankingTaskConfig.replace": true,
+    "tfr.keras.task.RankingTaskConfig.restrictions": true,
+    "tfr.keras.task.RankingTaskConfig.train_data": true,
+    "tfr.keras.task.RankingTaskConfig.validate": true,
+    "tfr.keras.task.RankingTaskConfig.validation_data": true,
     "tfr.keras.utils": false,
     "tfr.keras.utils.identity": false,
     "tfr.keras.utils.inverse": false,

tensorflow_ranking/g3doc/api_docs/python/tfr/_toc.yaml

@@ -231,6 +231,8 @@ toc:
       path: /ranking/api_docs/python/tfr/keras/model/PreprocessorWithSpec
     - title: Scorer
       path: /ranking/api_docs/python/tfr/keras/model/Scorer
+    - title: TensorDict
+      path: /ranking/api_docs/python/tfr/keras/model/TensorDict
     - title: TensorLike
       path: /ranking/api_docs/python/tfr/keras/model/TensorLike
     - title: TypeSpecInputCreator
@@ -295,6 +297,20 @@ toc:
       path: /ranking/api_docs/python/tfr/keras/strategy_utils/get_strategy
     - title: strategy_scope
       path: /ranking/api_docs/python/tfr/keras/strategy_utils/strategy_scope
+  - title: task
+    section:
+    - title: Overview
+      path: /ranking/api_docs/python/tfr/keras/task
+    - title: FeatureSpec
+      path: /ranking/api_docs/python/tfr/keras/task/FeatureSpec
+    - title: RankingDataConfig
+      path: /ranking/api_docs/python/tfr/keras/task/RankingDataConfig
+    - title: RankingDataLoader
+      path: /ranking/api_docs/python/tfr/keras/task/RankingDataLoader
+    - title: RankingTask
+      path: /ranking/api_docs/python/tfr/keras/task/RankingTask
+    - title: RankingTaskConfig
+      path: /ranking/api_docs/python/tfr/keras/task/RankingTaskConfig
   - title: utils
     section:
     - title: Overview

tensorflow_ranking/g3doc/api_docs/python/tfr/all_symbols.md

@@ -104,6 +104,7 @@
 *   <a href="../tfr/keras/model/Preprocessor.md"><code>tfr.keras.model.Preprocessor</code></a>
 *   <a href="../tfr/keras/model/PreprocessorWithSpec.md"><code>tfr.keras.model.PreprocessorWithSpec</code></a>
 *   <a href="../tfr/keras/model/Scorer.md"><code>tfr.keras.model.Scorer</code></a>
+*   <a href="../tfr/keras/model/TensorDict.md"><code>tfr.keras.model.TensorDict</code></a>
 *   <a href="../tfr/keras/model/TensorLike.md"><code>tfr.keras.model.TensorLike</code></a>
 *   <a href="../tfr/keras/model/TypeSpecInputCreator.md"><code>tfr.keras.model.TypeSpecInputCreator</code></a>
 *   <a href="../tfr/keras/model/UnivariateScorer.md"><code>tfr.keras.model.UnivariateScorer</code></a>
@@ -132,6 +133,12 @@
 *   <a href="../tfr/keras/strategy_utils/get_output_filepath.md"><code>tfr.keras.strategy_utils.get_output_filepath</code></a>
 *   <a href="../tfr/keras/strategy_utils/get_strategy.md"><code>tfr.keras.strategy_utils.get_strategy</code></a>
 *   <a href="../tfr/keras/strategy_utils/strategy_scope.md"><code>tfr.keras.strategy_utils.strategy_scope</code></a>
+*   <a href="../tfr/keras/task.md"><code>tfr.keras.task</code></a>
+*   <a href="../tfr/keras/task/FeatureSpec.md"><code>tfr.keras.task.FeatureSpec</code></a>
+*   <a href="../tfr/keras/task/RankingDataConfig.md"><code>tfr.keras.task.RankingDataConfig</code></a>
+*   <a href="../tfr/keras/task/RankingDataLoader.md"><code>tfr.keras.task.RankingDataLoader</code></a>
+*   <a href="../tfr/keras/task/RankingTask.md"><code>tfr.keras.task.RankingTask</code></a>
+*   <a href="../tfr/keras/task/RankingTaskConfig.md"><code>tfr.keras.task.RankingTaskConfig</code></a>
 *   <a href="../tfr/keras/utils.md"><code>tfr.keras.utils</code></a>
 *   <a href="../tfr/keras/utils/identity.md"><code>tfr.keras.utils.identity</code></a>
 *   <a href="../tfr/keras/utils/inverse.md"><code>tfr.keras.utils.inverse</code></a>

tensorflow_ranking/g3doc/api_docs/python/tfr/data/build_ranking_dataset.md

@@ -11,7 +11,7 @@ description: Builds a ranking tf.dataset with a standard data format.
 
 <table class="tfo-notebook-buttons tfo-api nocontent" align="left">
 <td>
-  <a target="_blank" href="https://github.com/tensorflow/ranking/tree/master/tensorflow_ranking/python/data.py#L954-L1002">
+  <a target="_blank" href="https://github.com/tensorflow/ranking/tree/master/tensorflow_ranking/python/data.py#L996-L1044">
     <img src="https://www.tensorflow.org/images/GitHub-Mark-32px.png" />
     View source on GitHub
   </a>

tensorflow_ranking/g3doc/api_docs/python/tfr/data/build_ranking_dataset_with_parsing_fn.md

@@ -11,7 +11,7 @@ description: Builds a ranking tf.dataset using the provided parsing_fn.
 
 <table class="tfo-notebook-buttons tfo-api nocontent" align="left">
 <td>
-  <a target="_blank" href="https://github.com/tensorflow/ranking/tree/master/tensorflow_ranking/python/data.py#L857-L951">
+  <a target="_blank" href="https://github.com/tensorflow/ranking/tree/master/tensorflow_ranking/python/data.py#L899-L993">
     <img src="https://www.tensorflow.org/images/GitHub-Mark-32px.png" />
     View source on GitHub
   </a>

tensorflow_ranking/g3doc/api_docs/python/tfr/data/build_ranking_serving_input_receiver_fn.md

@@ -11,7 +11,7 @@ description: Returns a serving input receiver fn for a standard data format.
 
 <table class="tfo-notebook-buttons tfo-api nocontent" align="left">
 <td>
-  <a target="_blank" href="https://github.com/tensorflow/ranking/tree/master/tensorflow_ranking/python/data.py#L1035-L1079">
+  <a target="_blank" href="https://github.com/tensorflow/ranking/tree/master/tensorflow_ranking/python/data.py#L1077-L1121">
     <img src="https://www.tensorflow.org/images/GitHub-Mark-32px.png" />
     View source on GitHub
   </a>

tensorflow_ranking/g3doc/api_docs/python/tfr/data/build_ranking_serving_input_receiver_fn_with_parsing_fn.md

@@ -11,7 +11,7 @@ description: Returns a receiver function with the provided parsing_fn.
 
 <table class="tfo-notebook-buttons tfo-api nocontent" align="left">
 <td>
-  <a target="_blank" href="https://github.com/tensorflow/ranking/tree/master/tensorflow_ranking/python/data.py#L1005-L1032">
+  <a target="_blank" href="https://github.com/tensorflow/ranking/tree/master/tensorflow_ranking/python/data.py#L1047-L1074">
     <img src="https://www.tensorflow.org/images/GitHub-Mark-32px.png" />
     View source on GitHub
   </a>

tensorflow_ranking/g3doc/api_docs/python/tfr/data/build_sequence_example_serving_input_receiver_fn.md

@@ -11,7 +11,7 @@ description: Creates a serving_input_receiver_fn for SequenceExample inputs.
 
 <table class="tfo-notebook-buttons tfo-api nocontent" align="left">
 <td>
-  <a target="_blank" href="https://github.com/tensorflow/ranking/tree/master/tensorflow_ranking/python/data.py#L1248-L1279">
+  <a target="_blank" href="https://github.com/tensorflow/ranking/tree/master/tensorflow_ranking/python/data.py#L1290-L1321">
     <img src="https://www.tensorflow.org/images/GitHub-Mark-32px.png" />
     View source on GitHub
   </a>

tensorflow_ranking/g3doc/api_docs/python/tfr/data/build_tf_example_serving_input_receiver_fn.md

@@ -11,7 +11,7 @@ description: Builds a serving input fn for tensorflow.training.Example.
 
 <table class="tfo-notebook-buttons tfo-api nocontent" align="left">
 <td>
-  <a target="_blank" href="https://github.com/tensorflow/ranking/tree/master/tensorflow_ranking/python/data.py#L1332-L1369">
+  <a target="_blank" href="https://github.com/tensorflow/ranking/tree/master/tensorflow_ranking/python/data.py#L1374-L1411">
     <img src="https://www.tensorflow.org/images/GitHub-Mark-32px.png" />
     View source on GitHub
   </a>

tensorflow_ranking/g3doc/api_docs/python/tfr/data/make_parsing_fn.md

@@ -11,7 +11,7 @@ description: Returns a parsing fn for a standard data format.
 
 <table class="tfo-notebook-buttons tfo-api nocontent" align="left">
 <td>
-  <a target="_blank" href="https://github.com/tensorflow/ranking/tree/master/tensorflow_ranking/python/data.py#L800-L854">
+  <a target="_blank" href="https://github.com/tensorflow/ranking/tree/master/tensorflow_ranking/python/data.py#L842-L896">
     <img src="https://www.tensorflow.org/images/GitHub-Mark-32px.png" />
     View source on GitHub
   </a>