model_builder_tf2_test.py test file fails

I installed the tf od api as per the documentation and everything regarding the setup runs without error. But when i try to run the model_builder_tf2_test.py file to test my installation it fails at last step. Below is the error which I get. I even tried to create a fresh env but still it fails at the same step.
M using Linux Ubuntu 18.04.4 as CLI.

`
ERROR: test_create_center_net_model (main.ModelBuilderTF2Test)
ModelBuilderTF2Test.test_create_center_net_model
Test building a CenterNet model from proto txt.

Traceback (most recent call last):
File "object_detection/builders/model_builder_tf2_test.py", line 224, in test_create_center_net_model
model = model_builder.build(config, is_training=True)
File "/usr/local/lib/python3.6/dist-packages/object_detection/builders/model_builder.py", line 1090, in build
add_summaries)
File "/usr/local/lib/python3.6/dist-packages/object_detection/builders/model_builder.py", line 1040, in _build_center_net_model
compute_heatmap_sparse=center_net_config.compute_heatmap_sparse)
AttributeError: compute_heatmap_sparse
`