From d482382a92154492c0ac62f9ef086e1783cc7fa6 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Mon, 18 May 2026 06:39:41 -0700
Subject: [PATCH] Sync pyproject.toml fixes from main into pip (#5555)

- Add missing cu124onlytorch240 extras (xformers==0.0.28.post1)
- Add sentence-transformers to huggingface and colab-new extras
- Add triton extras group (split out from main deps)
- Add windows extras group
- Drop xformers<0.0.27 cap in colab-no-deps; gate on linux/win + x86_64
- Add [tool.ruff] config (target py311, lint select/ignore, extend-exclude)
- Add [tool.pytest.ini_options] testpaths = ["tests/security"]

Only PyPI-compatible (version-pin) entries copied; URL-based intel/amd/rocm/
flashattentiontorch* extras from main are intentionally skipped.
---
 pyproject.toml | 51 +++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 50 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 457178baba..182118a3fd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -86,6 +86,10 @@ include = ["unsloth*", "unsloth_cli*", "studio", "studio.backend*"]
 exclude = ["images*", "tests*", "*.node_modules", "*.node_modules.*"]
 
 [project.optional-dependencies]
+triton = [
+    "triton>=3.0.0 ; ('linux' in sys_platform)",
+    "triton-windows ; (sys_platform == 'win32') and (platform_machine == 'AMD64' or platform_machine == 'x86_64')",
+]
 huggingface = [
     "unsloth_zoo>=2026.5.2",
     "wheel>=0.42.0",
@@ -105,11 +109,17 @@ huggingface = [
     "diffusers",
     "transformers>=4.51.3,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3,!=4.53.0,!=4.54.0,!=4.55.0,!=4.55.1,!=4.57.0,!=4.57.4,!=4.57.5,!=5.0.0,!=5.1.0,<=5.5.0",
     "trl>=0.18.2,!=0.19.0,<=0.24.0",
+    "sentence-transformers",
     "typer",
     "pydantic",
     "pyyaml",
     "nest-asyncio",
 ]
+windows = [
+    "unsloth[huggingface]",
+    "bitsandbytes>=0.45.5,!=0.46.0,!=0.48.0 ; (sys_platform == 'win32')",
+    "xformers>=0.0.22.post7 ; (sys_platform == 'win32')",
+]
 base = [
     "unsloth[huggingface]",
 ]
@@ -149,6 +159,9 @@ cu118onlytorch240 = [
 cu121onlytorch240 = [
     "xformers==0.0.27.post2 ; ('linux' in sys_platform or sys_platform == 'win32') and (platform_machine == 'AMD64' or platform_machine == 'x86_64')",
 ]
+cu124onlytorch240 = [
+    "xformers==0.0.28.post1 ; ('linux' in sys_platform or sys_platform == 'win32') and (platform_machine == 'AMD64' or platform_machine == 'x86_64')",
+]
 cu118onlytorch250 = [
     "xformers==0.0.28.post2 ; ('linux' in sys_platform or sys_platform == 'win32') and (platform_machine == 'AMD64' or platform_machine == 'x86_64')",
 ]
@@ -490,6 +503,7 @@ colab-new = [
     "hf_transfer",
     "bitsandbytes>=0.45.5,!=0.46.0,!=0.48.0",
     "unsloth[triton]",
+    "sentence-transformers",
     "typer",
     "pydantic",
     "pyyaml",
@@ -499,7 +513,7 @@ colab-no-deps = [
     "accelerate>=0.34.1",
     "trl>=0.18.2,!=0.19.0,<=0.24.0",
     "peft>=0.18.0",
-    "xformers<0.0.27",
+    "xformers ; ('linux' in sys_platform or sys_platform == 'win32') and (platform_machine == 'AMD64' or platform_machine == 'x86_64')",
     "bitsandbytes>=0.45.5,!=0.46.0,!=0.48.0",
     "protobuf",
 ]
@@ -739,3 +753,38 @@ cu130-ampere-torch2100 = [
 homepage = "https://unsloth.ai"
 documentation = "https://unsloth.ai/docs"
 repository = "https://github.com/unslothai/unsloth"
+
+[tool.ruff]
+target-version = "py311"
+force-exclude = true
+extend-exclude = [
+    "*chat_templates.py",
+    "*ollama_template_mappers.py",
+    "*_auto_install.py",
+    "*mapper.py",
+]
+
+[tool.ruff.lint]
+select = ["E9", "F63", "F7", "F82"]
+ignore = [
+    "E402",
+    "E722",
+    "F403",
+    "F405",
+    "F811",
+    "F821",
+    "F841",
+    "F401",
+    "E731",
+    "E741",
+    "F601",
+    "E712",
+]
+
+[tool.ruff.format]
+
+[tool.pytest.ini_options]
+# Narrow the default test discovery so `pytest` from the repo root
+# does NOT pick up the GPU-heavy tests under tests/python, tests/qlora,
+# etc. The CI security job runs `pytest tests/security` explicitly.
+testpaths = ["tests/security"]