diff --git a/pyproject.toml b/pyproject.toml index e2173d6811..b06131021a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -58,7 +58,8 @@ studio = [ ] [tool.setuptools.packages.find] -exclude = ["images*", "tests*", "kernels/moe*"] +include = ["unsloth*", "unsloth_cli*", "studio", "studio.backend*"] +exclude = ["images*", "tests*", "*.node_modules", "*.node_modules.*"] [project.optional-dependencies] triton = [