-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathpyproject.toml
More file actions
121 lines (113 loc) · 3.1 KB
/
pyproject.toml
File metadata and controls
121 lines (113 loc) · 3.1 KB
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
[project]
name = "nl2shell"
version = "0.1.0"
description = "Natural Language to Shell Command translation using fine-tuned Qwen3.5-0.8B. Convert plain English to bash/shell commands instantly."
readme = "README.md"
license = "MIT"
requires-python = ">=3.10"
keywords = [
"nl2bash",
"nl2shell",
"natural-language-to-shell",
"natural-language-to-bash",
"shell-commands",
"bash-generation",
"fine-tuning",
"qlora",
"qwen",
"qwen3.5",
"transformers",
"llm",
"command-line",
"terminal",
"edge-ai",
]
classifiers = [
"Development Status :: 4 - Beta",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: System :: Shells",
]
[project.urls]
Homepage = "https://github.com/aryateja2106/nl2shell"
Documentation = "https://github.com/aryateja2106/nl2shell#readme"
Repository = "https://github.com/aryateja2106/nl2shell"
"Bug Tracker" = "https://github.com/aryateja2106/nl2shell/issues"
"Model" = "https://huggingface.co/AryaYT/nl2shell-0.8b"
"Dataset" = "https://huggingface.co/datasets/AryaYT/nl2shell-training"
"Demo" = "https://huggingface.co/spaces/AryaYT/nl2shell-demo"
[dependency-groups]
train = [
"torch>=2.0",
"transformers>=4.40",
"datasets>=2.18",
"peft>=0.10",
"trl>=0.8",
"bitsandbytes>=0.43",
"accelerate>=0.28",
"huggingface-hub>=0.22",
]
eval = [
"torch>=2.0",
"transformers>=4.40",
"datasets>=2.18",
"huggingface-hub>=0.22",
"nltk>=3.8",
]
demo = [
"torch>=2.0",
"transformers>=4.40",
"gradio>=4.0",
"huggingface-hub>=0.22",
]
dev = [
"ruff>=0.15",
"ty>=0.0.21",
]
data = [
"datasets>=2.18",
"huggingface-hub>=0.22",
"pandas>=2.0",
]
[tool.ruff]
target-version = "py310"
line-length = 100
[tool.ruff.lint]
select = [
"E", # pycodestyle errors
"W", # pycodestyle warnings
"F", # pyflakes
"I", # isort
"N", # pep8-naming
"UP", # pyupgrade
"B", # flake8-bugbear
"S", # flake8-bandit (security)
"T20", # flake8-print (allow prints in scripts)
"SIM", # flake8-simplify
"RUF", # ruff-specific
]
ignore = [
"T201", # allow print() — these are training scripts
"S101", # allow assert — used in benchmarks
"S603", # allow subprocess calls — training setup
"S607", # allow partial executable path
"E501", # line length handled by formatter
"E402", # allow late imports — scripts pip-install before importing
]
[tool.ruff.lint.isort]
known-first-party = ["nl2shell"]
[tool.ruff.format]
quote-style = "double"
indent-style = "space"
[tool.ty.environment]
python-version = "3.10"
[tool.ty.rules]
# GPU/Colab dependencies not installed locally — suppress import errors
unresolved-import = "ignore"