diff --git a/setup.py b/setup.py
index 3deaed54e0..f651c27898 100644
--- a/setup.py
+++ b/setup.py
@@ -58,7 +58,7 @@ def get_optional_dependencies():
     version=HERE.joinpath("VERSION").read_text().strip(),
     packages=find_packages("src"),
     package_dir={"": "src"},
-    package_data={"": ["*.whl"]},
+    package_data={"": ["*.whl", "py.typed"]},
     py_modules=[os.path.splitext(os.path.basename(path))[0] for path in glob("src/*.py")],
     include_package_data=True,
     install_requires=get_dependencies(),
diff --git a/src/sagemaker/py.typed b/src/sagemaker/py.typed
new file mode 100644
index 0000000000..e69de29bb2