tensorflow_ranking/python/metrics.py (12 lines):
	- line 109: # TODO: Add mask argument for metric.compute() call
	- line 300: # TODO: Add mask argument for metric.compute() call
	- line 326: # TODO: Add mask argument for metric.compute() call
	- line 351: # TODO: Add mask argument for metric.compute() call
	- line 376: # TODO: Add mask argument for metric.compute() call.
	- line 408: # TODO: Add mask argument for metric.compute() call
	- line 433: # TODO: Add mask argument for metric.compute() call
	- line 472: # TODO: Add mask argument for metric.compute() call
	- line 504: # TODO: Add mask argument for metric.compute() call
	- line 550: # TODO: Add mask argument for metric.compute() call
	- line 577: # TODO: Add mask argument for metric.compute() call
	- line 616: # TODO: Add mask argument for metric.compute() call


tensorflow_ranking/python/keras/metrics.py (2 lines):
	- line 186: # TODO: Add mask argument for metric.compute() call
	- line 388: # TODO Add recall metrics to TF1 in another cl.


tensorflow_ranking/python/head.py (2 lines):
	- line 318: # TODO: Figure out a better way to set train_op_fn and optimizer
	- line 432: # TODO: Add the per-head loss loss/head_name to metrics.


tensorflow_ranking/python/keras/pipeline.py (2 lines):
	- line 870: # TODO: Define these bulky types as globals at the top.
	- line 940: # TODO: Remove defaults common in Estimator pipeline and here.


tensorflow_ranking/python/keras/model.py (2 lines):
	- line 60: # TODO: Support compatibility with TPUs.
	- line 859: # TODO: These parameters may be inferred from the call inputs.


tensorflow_ranking/python/metrics_impl.py (1 line):
	- line 487: # TODO: Consider to add a cap position topn + 1 when there is no


tensorflow_ranking/python/feature.py (1 line):
	- line 104: # TODO: Ensure only v2 Feature Columns are used.


tensorflow_ranking/extension/premade/tfrbert_task.py (1 line):
	- line 107: # TODO: Extend to more generic encoders in addition to BERT.


tensorflow_ranking/python/losses.py (1 line):
	- line 527: TODO: Add the lambda_weight support.


tensorflow_ranking/python/utils.py (1 line):
	- line 431: # TODO: Add checks to validate (ragged) shapes of input tensors.


tensorflow_ranking/python/losses_impl.py (1 line):
	- line 963: # TODO: try lambda_weights for circle loss.


tensorflow_ranking/python/model.py (1 line):
	- line 264: # TODO: Be more smart to infer is_valid.


tensorflow_ranking/python/keras/strategy_utils.py (1 line):
	- line 115: # TODO: integrate PSStrategy to pipeline.


tensorflow_ranking/extension/pipeline.py (1 line):
	- line 424: # TODO: supports for distributed training and evaluation.


tensorflow_ranking/python/keras/losses.py (1 line):
	- line 39: # TODO: Add support for circle loss and neural sort losses.


tensorflow_ranking/python/estimator.py (1 line):
	- line 683: # TODO: Attach the link to the paper.