Export QuantizationSimModel API in root package (#3668)

Signed-off-by: Kyunggeun Lee <[email protected]>
quic · Dec 18, 2024 · b67711c · b67711c
1 parent 5e9ede4
commit b67711c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/TrainingExtensions/torch/src/python/aimet_torch/__init__.py b/TrainingExtensions/torch/src/python/aimet_torch/__init__.py
@@ -40,3 +40,6 @@
 # For more information about explicit namespace packages,
 # see https://packaging.python.org/en/latest/guides/packaging-namespace-packages
 __path__ = __import__('pkgutil').extend_path(__path__, __name__)
+
+from .quantsim import QuantizationSimModel
+from . import nn