diff --git a/pyproject.toml b/pyproject.toml
index b99bfc2a9d..1f18363c78 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -211,3 +211,6 @@ unfixable = ["F401"]
 [tool.ruff.mccabe]
 # Unlike Flake8, default to a complexity level of 10.
 max-complexity = 10
+
+[tool.setuptools.package-data]
+flaml = ["py.typed"]
\ No newline at end of file