diff --git a/TrainingExtensions/torch/src/python/aimet_torch/v2/__init__.py b/TrainingExtensions/torch/src/python/aimet_torch/v2/__init__.py
index 97e8e3d7c74..1f23f88174f 100644
--- a/TrainingExtensions/torch/src/python/aimet_torch/v2/__init__.py
+++ b/TrainingExtensions/torch/src/python/aimet_torch/v2/__init__.py
@@ -34,8 +34,7 @@
 #
 #  @@-COPYRIGHT-END-@@
 # =============================================================================
-
-# pylint: disable=missing-docstring
+""" aimet_torch.v2 subpackage """
 from . import experimental
 from . import nn
 from . import quantization
diff --git a/TrainingExtensions/torch/src/python/aimet_torch/v2/quantization/__init__.py b/TrainingExtensions/torch/src/python/aimet_torch/v2/quantization/__init__.py
index 54ba7bf4021..b371f24035f 100644
--- a/TrainingExtensions/torch/src/python/aimet_torch/v2/quantization/__init__.py
+++ b/TrainingExtensions/torch/src/python/aimet_torch/v2/quantization/__init__.py
@@ -34,7 +34,8 @@
 #
 #  @@-COPYRIGHT-END-@@
 # =============================================================================
-
-# pylint: disable=all
-
+# pylint: disable=missing-docstring
 from .tensor import *
+from . import base
+from . import affine
+from . import float # pylint: disable=, redefined-builtin