Handle class balancing in a way to support TLT.
This pull request **fixes 3 alerts** when merging 2817d5e5b74e67436fcec1ff3adef8f9b138c265 into efaab9e67346d633e3efd49c44a5c2b83661d1fc - [view on LGTM.com](https://lgtm.com/projects/g/NVIDIA/NeMo/rev/pr-f1f11b89c657df4cd1e55fc67022c52b4bf50e66)

**fixed alerts:**

* 2 for Unused local variable
* 1 for Unused import This pull request **fixes 3 alerts** when merging fc7a45b0719139f60f0a62510c1f19b266940522 into 4d6cad6a4e59e9d10fcc82984c1ddb3a1f40067f - [view on LGTM.com](https://lgtm.com/projects/g/NVIDIA/NeMo/rev/pr-64d97a7b249c433648647052ef331e51826914f1)

**fixed alerts:**

* 2 for Unused local variable
* 1 for Unused import > 
> 
> Please rebase to latest main - export() and attach_onnx_to_onnx should disappear.

Dropped them. This pull request **fixes 2 alerts** when merging 2c32926718d5e73878a294594a449e316b6ed9bf into f4ffb4300d6ef1227e0a2cec9c692a38606be299 - [view on LGTM.com](https://lgtm.com/projects/g/NVIDIA/NeMo/rev/pr-a65e4ee17c4a44d729f0bcb33450c2768f21faec)

**fixed alerts:**

* 2 for Unused local variable This pull request **fixes 2 alerts** when merging 270225370b979757b56fc83aed66a5ec611e5ba7 into 4b74402feeeb672f8982bb9fa51861f6c1d5f94b - [view on LGTM.com](https://lgtm.com/projects/g/NVIDIA/NeMo/rev/pr-4cb7bb67a2c3b70709410a3d4600a565f7756d97)

**fixed alerts:**

* 2 for Unused local variable This pull request **introduces 4 alerts** and **fixes 3** when merging e960719173952bde00afb73f501fdf59ea89d049 into 89a8ea037a121001e105e95b49d71906a8de29aa - [view on LGTM.com](https://lgtm.com/projects/g/NVIDIA/NeMo/rev/pr-dff537058ff2f9f5e2402227e7a70b7901933ea1)

**new alerts:**

* 4 for Unused import

**fixed alerts:**

* 2 for Unused local variable
* 1 for Unused import This pull request **fixes 2 alerts** when merging e960719173952bde00afb73f501fdf59ea89d049 into e816905e286df545ce344e865ea6cf8e4b0c85ca - [view on LGTM.com](https://lgtm.com/projects/g/NVIDIA/NeMo/rev/pr-00578344bfd1f5b5828305d31be724d2517c5881)

**fixed alerts:**

* 2 for Unused local variable This pull request **fixes 2 alerts** when merging d975797c7625d62d8a94c808c3107c7d58cec3e7 into 89a8ea037a121001e105e95b49d71906a8de29aa - [view on LGTM.com](https://lgtm.com/projects/g/NVIDIA/NeMo/rev/pr-75901b1980139ecbd71de6d9fa50d2aa4d594d5b)

**fixed alerts:**

* 2 for Unused local variable