Python TensorFlow v2, locating contrib.training

Question

I am using some code based on Tensorflow 1.x and I am using the latest version of TensorFlow (2.8).

I have run tf_upgrade_v2 (following these migration instructions) on the python file to generate a new compatible file.

However, this doesn't work with the line of code; from TensorFlow.contrib import training as contrib_training.

Getting the error ModuleNotFoundError: No module named 'tensorflow.contrib'

I understand that contrib has been deprecated, so the modules have been moved around. But I haven't been able to find where this training module has moved to or if it has been deleted. I have looked for information first on this document on the status of tf.contrib and after this spreadsheet on TF2.0 Symbols map and haven't found any information on the fate of this training module;

I have also tried looking around TensforFlow/tf.compat.v1 but with no success.

Of course uninstalling this version of TensorFlow and instead installing an earlier version is an option. But not an option that I am overly keen on.

I feel like I may be missing something fairly obvious here...

Thanks in advance!!