As noted in [TODO(b/208879020)](https://github.com/tensorflow/transform/blob/ee1a769f0e359a8722dca7b434a3b499396a140f/examples/census_example_v2.py#L196) in [examples/census_example_v2.py](https://github.com/tensorflow/transform/blob/master/examples/census_example_v2.py), this needs to be updated. I have submitted [pr270](https://github.com/tensorflow/transform/pull/270) to address this.