saferax/pyproject.toml at master · krypticmouse/saferax · 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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
[tool.poetry]
name = "saferax"
version = "0.0.2"
description = "Loading Safetensor from HF to Equinox"
authors = ["Herumb Shandilya <[email protected]>"]
license = "Apache-2.0 license"
readme = "README.md"

[tool.poetry.dependencies]
python = ">=3.10"
jax = "^0.4.38"
equinox = "^0.11.11"
fsspec = "^2024.12.0"
huggingface-hub = "^0.27.0"
safetensors = "^0.4.5"
pytest = "^8.3.4"


[tool.poetry.group.dev.dependencies]
pyright = "^1.1.391"
ruff = "^0.8.4"


[tool.ruff]
line-length = 120
exclude = ["examples", "__init__.py"]

[tool.ruff.lint]
select = [
    "E",  # pycodestyle errors
    "W",  # pycodestyle warnings
    "F",  # pyflakes
    "I",  # isort
    "B",  # flake8-bugbear
    "C4",  # flake8-comprehensions
    "N",  # PEP8 naming convetions
]
ignore = [
    "C901",  # too complex
    "W191",  # indentation contains tabs
    "D401",  # imperative mood
    "I001",
    "N812",
    "N803",
]

[tool.ruff.lint.pydocstyle]
convention = "google"


[tool.pyright]
include = ["saferax"]
exclude = [
    "**/__pycache__",
]
defineConstant = { DEBUG = true }

reportMissingImports = "error"
reportMissingTypeStubs = false

pythonVersion = "3.10"
pythonPlatform = "Linux"


[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"