lint
This commit is contained in:
@@ -38,13 +38,13 @@ dependencies = [
|
|||||||
"hf_xet==1.1.5",
|
"hf_xet==1.1.5",
|
||||||
"kernels==0.9.0",
|
"kernels==0.9.0",
|
||||||
"trackio",
|
"trackio",
|
||||||
|
|
||||||
# Optimization and training
|
# Optimization and training
|
||||||
"optimum==1.16.2",
|
"optimum==1.16.2",
|
||||||
"hf_transfer",
|
"hf_transfer",
|
||||||
"sentencepiece",
|
"sentencepiece",
|
||||||
"gradio==5.41.1",
|
"gradio==5.41.1",
|
||||||
|
|
||||||
# Infrastructure
|
# Infrastructure
|
||||||
"modal==1.0.2",
|
"modal==1.0.2",
|
||||||
"pydantic==2.10.6",
|
"pydantic==2.10.6",
|
||||||
@@ -57,7 +57,7 @@ dependencies = [
|
|||||||
"colorama",
|
"colorama",
|
||||||
"numba",
|
"numba",
|
||||||
"numpy>=1.24.4,<3.0",
|
"numpy>=1.24.4,<3.0",
|
||||||
|
|
||||||
# QLora dependencies
|
# QLora dependencies
|
||||||
"evaluate==0.4.1",
|
"evaluate==0.4.1",
|
||||||
"scipy",
|
"scipy",
|
||||||
@@ -66,33 +66,33 @@ dependencies = [
|
|||||||
"art",
|
"art",
|
||||||
"tensorboard",
|
"tensorboard",
|
||||||
"python-dotenv==1.0.1",
|
"python-dotenv==1.0.1",
|
||||||
|
|
||||||
# Remote filesystems
|
# Remote filesystems
|
||||||
"s3fs>=2024.5.0",
|
"s3fs>=2024.5.0",
|
||||||
"gcsfs>=2024.5.0",
|
"gcsfs>=2024.5.0",
|
||||||
"adlfs>=2024.5.0",
|
"adlfs>=2024.5.0",
|
||||||
"ocifs==1.3.2",
|
"ocifs==1.3.2",
|
||||||
|
|
||||||
# Other utilities
|
# Other utilities
|
||||||
"zstandard>=0.23.0",
|
"zstandard>=0.23.0",
|
||||||
"fastcore",
|
"fastcore",
|
||||||
|
|
||||||
# LM eval harness
|
# LM eval harness
|
||||||
"lm_eval==0.4.7",
|
"lm_eval==0.4.7",
|
||||||
"langdetect==1.0.9",
|
"langdetect==1.0.9",
|
||||||
"immutabledict==4.2.0",
|
"immutabledict==4.2.0",
|
||||||
"antlr4-python3-runtime==4.13.2",
|
"antlr4-python3-runtime==4.13.2",
|
||||||
|
|
||||||
# Training optimizers
|
# Training optimizers
|
||||||
"schedulefree==1.4.1",
|
"schedulefree==1.4.1",
|
||||||
|
|
||||||
# Axolotl contribs
|
# Axolotl contribs
|
||||||
"axolotl-contribs-lgpl @ git+https://github.com/axolotl-ai-cloud/axolotl-contribs-lgpl.git@numpy",
|
"axolotl-contribs-lgpl @ git+https://github.com/axolotl-ai-cloud/axolotl-contribs-lgpl.git@numpy",
|
||||||
"axolotl-contribs-mit==0.0.5",
|
"axolotl-contribs-mit==0.0.5",
|
||||||
|
|
||||||
# Mistral
|
# Mistral
|
||||||
"mistral-common==1.8.3",
|
"mistral-common==1.8.3",
|
||||||
|
|
||||||
# Platform-specific dependencies (Linux by default, excluded on macOS)
|
# Platform-specific dependencies (Linux by default, excluded on macOS)
|
||||||
"triton>=3.0.0,<3.4.0 ; sys_platform != 'darwin'",
|
"triton>=3.0.0,<3.4.0 ; sys_platform != 'darwin'",
|
||||||
"xformers>=0.0.23.post1 ; sys_platform != 'darwin'",
|
"xformers>=0.0.23.post1 ; sys_platform != 'darwin'",
|
||||||
@@ -199,8 +199,31 @@ line-length = 88
|
|||||||
target-version = "py310"
|
target-version = "py310"
|
||||||
|
|
||||||
[tool.ruff.lint]
|
[tool.ruff.lint]
|
||||||
select = ["E", "F", "I", "W"]
|
select = ["E", "F", "W", "C90", "B"]
|
||||||
ignore = ["E501"]
|
ignore = [
|
||||||
|
"E203", # Whitespace before ':'
|
||||||
|
"E501", # Line too long
|
||||||
|
"C901", # Too complex
|
||||||
|
"B019", # Use of functools.cache on methods
|
||||||
|
"E722", # Bare except
|
||||||
|
"F821", # Undefined name (for dynamic exec)
|
||||||
|
]
|
||||||
|
|
||||||
|
[tool.ruff.lint.isort]
|
||||||
|
known-third-party = ["wandb", "comet_ml"]
|
||||||
|
known-local-folder = ["src", "tests"]
|
||||||
|
# Black-compatible isort settings
|
||||||
|
force-single-line = false
|
||||||
|
combine-as-imports = true
|
||||||
|
split-on-trailing-comma = true
|
||||||
|
|
||||||
|
[tool.ruff.format]
|
||||||
|
# Use black's formatting style exactly
|
||||||
|
quote-style = "double"
|
||||||
|
indent-style = "space"
|
||||||
|
skip-magic-trailing-comma = false
|
||||||
|
line-ending = "auto"
|
||||||
|
docstring-code-format = false
|
||||||
|
|
||||||
[tool.mypy]
|
[tool.mypy]
|
||||||
python_version = "3.10"
|
python_version = "3.10"
|
||||||
|
|||||||
Reference in New Issue
Block a user