OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00
Code Issues Actions 52 Packages Projects Releases Wiki Activity
Files
b870068d3de770b2bf2f6abbf86327f7b8e1ef3b
pytorch/pyproject.toml
albanD 5a7a65a4d5 Refine S101 skips (#169715) 
Next step towards removing asserts
Pull Request resolved: https://github.com/pytorch/pytorch/pull/169715
Approved by: https://github.com/Skylion007
2025-12-08 18:45:06 +00:00

476 lines
14 KiB
TOML
Raw Blame History

# Package ######################################################################
[build-system]
requires = [
# 70.1.0: min version for integrated bdist_wheel command from wheel package
# 77.0.0: min version for SPDX expression support for project.license
"setuptools>=70.1.0",
"cmake>=3.27",
"ninja",
"numpy",
"packaging",
"pyyaml",
"requests",
"six", # dependency chain: NNPACK -> PeachPy -> six
"typing-extensions>=4.10.0",
]
build-backend = "setuptools.build_meta"
[dependency-groups]
dev = [
# This list should be kept in sync with the requirements-build.txt
# in PyTorch root until the project fully migrates to pyproject.toml
# after which this can be removed as it is already specified in the
# [build-system] section
"setuptools>=70.1.0,<80.0", # setuptools develop deprecated on 80.0
"cmake>=3.27",
"ninja",
"numpy",
"packaging",
"pyyaml",
"requests",
"six", # dependency chain: NNPACK -> PeachPy -> six
"typing-extensions>=4.10.0",
# This list should be kept in sync with the requirements.txt in
# PyTorch root until the project fully migrates to pyproject.toml
"build[uv]",
"expecttest>=0.3.0",
"filelock",
"fsspec>=0.8.5",
"hypothesis",
"jinja2",
"lintrunner; platform_machine != 's390x' and platform_machine != 'riscv64'",
"networkx>=2.5.1",
"optree>=0.13.0",
"psutil",
"sympy>=1.13.3",
"typing-extensions>=4.15.0",
"wheel",
]
[project]
name = "torch"
description = "Tensors and Dynamic neural networks in Python with strong GPU acceleration"
readme = "README.md"
requires-python = ">=3.10"
# TODO: change to `license = "BSD-3-Clause"` and enable PEP 639 after pinning setuptools>=77
# FIXME: As of 2025.06.20, it is hard to ensure the minimum version of setuptools in our CI environment.
# TOML-table-based license deprecated in setuptools>=77, and the deprecation warning will be changed
# to an error on 2026.02.18. See also: https://github.com/pypa/setuptools/issues/4903
license = { text = "BSD-3-Clause" }
authors = [{ name = "PyTorch Team", email = "packages@pytorch.org" }]
keywords = ["pytorch", "machine learning"]
classifiers = [
"Development Status :: 5 - Production/Stable",
"Intended Audience :: Developers",
"Intended Audience :: Education",
"Intended Audience :: Science/Research",
"Topic :: Scientific/Engineering",
"Topic :: Scientific/Engineering :: Mathematics",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Software Development",
"Topic :: Software Development :: Libraries",
"Topic :: Software Development :: Libraries :: Python Modules",
"Programming Language :: C++",
"Programming Language :: Python :: 3 :: Only",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
"Programming Language :: Python :: 3.13",
"Programming Language :: Python :: 3.14",
]
dynamic = [
"entry-points",
"dependencies",
"scripts",
"version",
]
[project.urls]
Homepage = "https://pytorch.org"
Repository = "https://github.com/pytorch/pytorch"
Documentation = "https://pytorch.org/docs"
"Issue Tracker" = "https://github.com/pytorch/pytorch/issues"
Forum = "https://discuss.pytorch.org"
[project.optional-dependencies]
optree = ["optree>=0.13.0"]
opt-einsum = ["opt-einsum>=3.3"]
pyyaml = ["pyyaml"]
# Linter tools #################################################################
[tool.isort]
src_paths = ["caffe2", "torch", "torchgen", "functorch", "test"]
extra_standard_library = ["typing_extensions"]
skip_gitignore = true
skip_glob = ["third_party/*"]
atomic = true
profile = "black"
indent = 4
line_length = 88
lines_after_imports = 2
multi_line_output = 3
include_trailing_comma = true
combine_as_imports = true
[tool.usort.known]
first_party = ["caffe2", "torch", "torchgen", "functorch", "test"]
standard_library = ["typing_extensions"]
[tool.ruff]
line-length = 88
src = ["caffe2", "torch", "torchgen", "functorch", "test"]
[tool.ruff.format]
docstring-code-format = true
quote-style = "double"
[tool.ruff.lint]
# NOTE: Synchoronize the ignores with .flake8
external = [
"B001",
"B902",
"B950",
"E121",
"E122",
"E128",
"E131",
"E704",
"E723",
"F723",
"F812",
"P201",
"P204",
"T484",
"TOR901",
]
ignore = [
# these ignores are from flake8-bugbear; please fix!
"B007", "B008", "B017",
"B018", # Useless expression
"B023",
"B028", # No explicit `stacklevel` keyword argument found
"E402",
"C408", # C408 ignored because we like the dict keyword argument syntax
"E501", # E501 is not flexible enough, we're using B950 instead
"E741",
"EXE001",
"F405",
"FURB122", # writelines
# these ignores are from ruff NPY; please fix!
"NPY002",
# these ignores are from ruff PERF; please fix!
"PERF203",
"PERF401",
# these ignores are from PYI; please fix!
"PYI024",
"PYI036",
"PYI041",
"PYI056",
"SIM102", "SIM103", "SIM112", # flake8-simplify code styles
"SIM105", # these ignores are from flake8-simplify. please fix or ignore with commented reason
"SIM108", # SIM108 ignored because we prefer if-else-block instead of ternary expression
"SIM110", # Checks for for loops that can be replaced with a builtin function, like any or all.
"SIM114", # Combine `if` branches using logical `or` operator
"SIM116", # Disable Use a dictionary instead of consecutive `if` statements
"SIM117",
"SIM300", # Yoda condition detected
"UP007", # keep-runtime-typing
"UP045", # keep-runtime-typing
"TC006",
# TODO: Remove Python-3.10 specific suppressions
"B905",
]
select = [
"B",
"B904", # Re-raised error without specifying the cause via the from keyword
"C4",
"G",
"E",
"EXE",
"F",
"SIM",
"W",
# Not included in flake8
"FURB",
"LOG",
"NPY",
"PERF",
"PGH004",
"PIE",
"PLC0131", # type bivariance
"PLC0132", # type param mismatch
"PLC1802", # len({expression}) used as condition without comparison
"PLC0205", # string as __slots__
"PLC3002", # unnecessary-direct-lambda-call
"PLC0414", # Import alias does not rename original package
"PLE",
"PLR0133", # constant comparison
"PLR0206", # property with params
"PLR1722", # use sys exit
"PLR1736", # unnecessary list index
"PLW0127", # Self-assignment of variable
"PLW0129", # assert on string literal
"PLW0131", # named expr without context
"PLW0133", # useless exception statement
"PLW0245", # super without brackets
"PLW0406", # import self
"PLW0711", # binary op exception
"PLW1501", # bad open mode
"PLW1507", # shallow copy os.environ
"PLW1509", # preexec_fn not safe with threads
"PLW2101", # useless lock statement
"PLW3301", # nested min max
"PT006", # TODO: enable more PT rules
"PT014", # duplicate parameterize case
"PT022",
"PT023",
"PT024",
"PT025",
"PT026",
"PYI",
"Q003", # avoidable escaped quote
"Q004", # unnecessary escaped quote
"RSE",
"RUF007", # pairwise over zip
"RUF008", # mutable dataclass default
"RUF013", # ban implicit optional
"RUF015", # access first ele in constant time
"RUF016", # type error non-integer index
"RUF017",
"RUF018", # no assignment in assert
"RUF019", # unnecessary-key-check
"RUF020", # never union
"RUF024", # from keys mutable
"RUF026", # default factory kwarg
"RUF030", # No print statement in assert
"RUF033", # default values __post_init__ dataclass
"RUF041", # simplify nested Literal
"RUF048", # properly parse `__version__`
"RUF200", # validate pyproject.toml
"S324", # for hashlib FIPS compliance
"SLOT",
"TC",
"TRY002", # ban vanilla raise (todo fix NOQAs)
"TRY203",
"TRY401", # verbose-log-message
"UP",
"YTT",
"S101",
]
[tool.ruff.lint.pyupgrade]
# Preserve types, even if a file imports `from __future__ import annotations`.
keep-runtime-typing = true
[tool.ruff.lint.per-file-ignores]
"__init__.py" = [
"F401",
]
"*.pyi" = [
"PYI011", # typed-argument-default-in-stub
"PYI021", # docstring-in-stub
"PYI053", # string-or-bytes-too-long
]
"functorch/docs/source/tutorials/**" = [
"F401",
]
"test/export/**" = [
"PGH004"
]
"test/typing/**" = [
"PGH004"
]
"test/typing/reveal/**" = [
"F821",
]
"test/torch_np/numpy_tests/**" = [
"F821",
"NPY201",
]
"test/dynamo/test_bytecode_utils.py" = [
"F821",
]
"test/dynamo/test_debug_utils.py" = [
"UP037",
]
"test/dynamo/test_misc.py" = [
"PGH004",
]
"test/jit/**" = [
"PLR0133", # tests require this for JIT
"PYI",
"RUF015",
"UP", # We don't want to modify the jit test as they test specify syntax
]
"test/test_jit.py" = [
"PLR0133", # tests require this for JIT
"PYI",
"RUF015",
"UP", # We don't want to modify the jit test as they test specify syntax
]
"test/inductor/s429861_repro.py" = [
"PGH004",
]
"test/inductor/test_torchinductor.py" = [
"UP037",
]
# autogenerated #TODO figure out why file level noqa is ignored
"torch/_appdirs.py" = ["PGH004"]
"torch/jit/_shape_functions.py" = ["PGH004"]
"torch/_inductor/fx_passes/serialized_patterns/**" = ["F401", "F501"]
"torch/_inductor/autoheuristic/artifacts/**" = ["F401", "F501"]
"torch/_inductor/codegen/**" = [
"PGH004"
]
"torchgen/api/types/__init__.py" = [
"F401",
"F403",
]
"torch/utils/collect_env.py" = [
"UP", # collect_env.py needs to work with older versions of Python
]
"torch/_vendor/**" = [
"UP", # No need to mess with _vendor
]
"tools/linter/**" = [
"LOG015" # please fix
]
"benchmarks/**" = [
"S101"
]
"test/**" = [
"S101"
]
"torchgen/**" = [
"S101"
]
"torch/__init__.py" = ["S101"]
"torch/_custom_op/**" = ["S101"]
"torch/_decomp/**" = ["S101"]
"torch/_dispatch/python.py" = ["S101"]
"torch/_dynamo/**" = ["S101"]
"torch/_export/**" = ["S101"]
"torch/_functorch/**" = ["S101"]
"torch/_guards.py" = ["S101"]
"torch/_higher_order_ops/**" = ["S101"]
"torch/_inductor/**" = ["S101"]
"torch/_jit_internal.py" = ["S101"]
"torch/_lazy/extract_compiled_graph.py" = ["S101"]
"torch/_library/**" = ["S101"]
"torch/_lobpcg.py" = ["S101"]
"torch/_logging/_internal.py" = ["S101"]
"torch/_meta_registrations.py" = ["S101"]
"torch/_numpy/_dtypes_impl.py" = ["S101"]
"torch/_numpy/_funcs_impl.py" = ["S101"]
"torch/_numpy/_ndarray.py" = ["S101"]
"torch/_numpy/_reductions_impl.py" = ["S101"]
"torch/_numpy/_util.py" = ["S101"]
"torch/_numpy/testing/**" = ["S101"]
"torch/_ops.py" = ["S101"]
"torch/_prims/**" = ["S101"]
"torch/_prims_common/**" = ["S101"]
"torch/_refs/__init__.py" = ["S101"]
"torch/_refs/fft.py" = ["S101"]
"torch/_refs/special/**" = ["S101"]
"torch/_subclasses/**" = ["S101"]
"torch/_tensor.py" = ["S101"]
"torch/_tensor_str.py" = ["S101"]
"torch/_utils.py" = ["S101"]
"torch/_weights_only_unpickler.py" = ["S101"]
"torch/amp/**" = ["S101"]
"torch/ao/nn/**" = ["S101"]
"torch/ao/pruning/_experimental/data_sparsifier/lightning/tests/test_callbacks.py" = ["S101"]
"torch/ao/quantization/pt2e/utils.py" = ["S101"]
"torch/backends/_coreml/**" = ["S101"]
"torch/backends/_nnapi/**" = ["S101"]
"torch/backends/mkl/**" = ["S101"]
"torch/backends/mkldnn/**" = ["S101"]
"torch/backends/xeon/**" = ["S101"]
"torch/compiler/_cache.py" = ["S101"]
"torch/cuda/**" = ["S101"]
"torch/distributed/**" = ["S101"]
"torch/distributions/**" = ["S101"]
"torch/export/**" = ["S101"]
"torch/functional.py" = ["S101"]
"torch/futures/**" = ["S101"]
"torch/fx/**" = ["S101"]
"torch/hub.py" = ["S101"]
"torch/jit/**" = ["S101"]
"torch/library.py" = ["S101"]
"torch/masked/_ops.py" = ["S101"]
"torch/mtia/mtia_graph.py" = ["S101"]
"torch/multiprocessing/__init__.py" = ["S101"]
"torch/nativert/backends/_lower_utils.py" = ["S101"]
"torch/nested/**" = ["S101"]
"torch/nn/**" = ["S101"]
"torch/onnx/_internal/**" = ["S101"]
"torch/onnx/ops/_symbolic_impl.py" = ["S101"]
"torch/overrides.py" = ["S101"]
"torch/package/_mangling.py" = ["S101"]
"torch/package/_package_pickler.py" = ["S101"]
"torch/package/_stdlib.py" = ["S101"]
"torch/package/analyze/find_first_use_of_broken_modules.py" = ["S101"]
"torch/package/importer.py" = ["S101"]
"torch/package/package_exporter.py" = ["S101"]
"torch/package/package_importer.py" = ["S101"]
"torch/profiler/_memory_profiler.py" = ["S101"]
"torch/profiler/_pattern_matcher.py" = ["S101"]
"torch/profiler/_utils.py" = ["S101"]
"torch/profiler/python_tracer.py" = ["S101"]
"torch/quantization/_quantized_conversions.py" = ["S101"]
"torch/serialization.py" = ["S101"]
"torch/sparse/**" = ["S101"]
"torch/storage.py" = ["S101"]
"torch/testing/**" = ["S101"]
"torch/utils/_debug_mode.py" = ["S101"]
"torch/utils/_python_dispatch.py" = ["S101"]
"torch/utils/_pytree.py" = ["S101"]
"torch/utils/_runtime_estimation.py" = ["S101"]
"torch/utils/cpp_extension.py" = ["S101"]
"tools/**" = [
"S101"
]
"setup.py" = [
"S101"
]
"functorch/**" = [
"S101"
]
"docs/source/scripts/exportdb/generate_example_rst.py" = ["S101"]
"android/pytorch_android/generate_test_torchscripts.py" = ["S101"]
".github/scripts/close_nonexistent_disable_issues.py" = ["S101"]
".github/scripts/delete_old_branches.py" = ["S101"]
".github/scripts/get_workflow_job_id.py" = ["S101"]
".github/scripts/gitutils.py" = ["S101"]
".github/scripts/label_utils.py" = ["S101"]
".github/scripts/test_trymerge.py" = ["S101"]
".github/scripts/trymerge.py" = ["S101"]
".ci/docker/manywheel/build_scripts/manylinux1-check.py" = ["S101"]
".ci/pytorch/smoke_test/check_binary_symbols.py" = ["S101"]
".ci/pytorch/smoke_test/smoke_test.py" = ["S101"]
".ci/lumen_cli/tests/test_vllm.py" = ["S101"]
".ci/lumen_cli/tests/test_run_plan.py" = ["S101"]
[tool.codespell]
ignore-words = "tools/linter/dictionary.txt"
[tool.spin]
package = 'torch'
[tool.spin.commands]
"Build" = [
".spin/cmds.py:lint",
".spin/cmds.py:fixlint",
".spin/cmds.py:quicklint",
".spin/cmds.py:quickfix",
]
"Regenerate" = [
".spin/cmds.py:regenerate_version",
".spin/cmds.py:regenerate_type_stubs",
".spin/cmds.py:regenerate_clangtidy_files",
".spin/cmds.py:regenerate_github_workflows",
]
Reference in New Issue View Git Blame Copy Permalink