Skip to content

Commit 476c867

Browse files
authored
Fix issues with dependencies not installing from git (#330)
1 parent 75c761a commit 476c867

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

pyproject.toml

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,9 @@ backend = [
1919
"hf-xet>=1.1.0",
2020
"bitsandbytes>=0.45.2",
2121
"unsloth==2025.8.1",
22-
"unsloth-zoo",
22+
"unsloth-zoo @ git+https://github.com/bradhilton/unsloth-zoo@323cf5e",
2323
"vllm==0.10.0",
24-
"torchtune",
24+
"torchtune @ git+https://github.com/pytorch/torchtune.git@2344509cf83bd886538fe3e8263e5145d1afb5c2",
2525
"trl>=0.19.0",
2626
"torch>=2.7.0",
2727
"torchao>=0.9.0",
@@ -61,7 +61,7 @@ sources = ["src"]
6161
[tool.hatch.build.targets.sdist]
6262
exclude = [
6363
"/dev",
64-
"/wandb",
64+
"/wandb",
6565
"/.art",
6666
"/.ruff_cache",
6767
"/.venv",
@@ -94,6 +94,4 @@ dev-dependencies = [
9494

9595
[tool.uv.sources]
9696
panza = { git = "https://github.com/corbt/panza.git" }
97-
torchtune = { git = "https://github.com/pytorch/torchtune.git", rev = "2344509cf83bd886538fe3e8263e5145d1afb5c2" }
9897
sweagent = { git = "https://github.com/bradhilton/SWE-agent" }
99-
unsloth-zoo = { git = "https://github.com/bradhilton/unsloth-zoo" }

src/art/dev/engine.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
1-
from typing import Any, Literal, Tuple, TypedDict
1+
from typing import Any, Literal, Tuple
2+
3+
from typing_extensions import TypedDict
24

35

46
class EngineArgs(TypedDict, total=False):

src/art/dev/torchtune.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
from typing import Literal, TypedDict
1+
from typing import Literal
22

3-
from typing_extensions import Required
3+
from typing_extensions import Required, TypedDict
44

55
# Update from torchtune.models.{model_family}.__init__.py files
66
# Only include argument-less TransformerDecoder factory functions

0 commit comments

Comments
 (0)