swe-gym/pyproject.toml at main · webcoderz/swe-gym · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
[project]
name = "swe-gym"
version = "0.1.0"
description = "SWE-Smith RL gym + GEPA gskill optimization pipeline to fine-tune a model or create skills for a coding agent on any repository."
requires-python = ">=3.10,<3.13"
dependencies = [
    "swesmith[all]",
    "gepa @ git+https://github.com/gepa-ai/gepa.git@1e7bfe295cd5046d6bc580b38277247502ad1f7e",
    "swebench",
    "mini-swe-agent",
    "pyyaml>=6.0",
    "cloudpickle>=3.0",
    "datasets>=3.0",
    "wandb>=0.18",
]

[tool.setuptools]
packages = []

[project.optional-dependencies]
sft = [
    "torchtune",
    "torch",
    "torchao",
]
unsloth = [
    "unsloth",
    "torch",
    "trl",
    "datasets",
    "optuna>=4.0",
]
dev = [
    "pytest>=8.0",
    "ruff>=0.4",
]

[tool.ruff]
line-length = 120

[tool.ruff.lint]
select = ["E", "F", "W"]
ignore = ["E501"]  # Line length handled by formatter