trex/pyproject.toml at main · apoorvalal/trex · 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
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
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "trex"
version = "0.1.0"
dependencies = [
  "torch>=2.0.0",
  "numpy>=1.20.0",
  "pytorch-minimize",
]
requires-python = ">=3.10"
authors = [
  {name = "Apoorva Lal", email = "lal.apoorva@gmail.com"},
]
description = "GPU-accelerated econometrics in PyTorch."
readme = "README.md"
license = "MIT"
keywords = ["econometrics", "pytorch", "gpu", "machine learning", "statistics", "fixed-effects"]
classifiers = [
  "Development Status :: 4 - Beta",
  "Programming Language :: Python :: 3",
  "Programming Language :: Python :: 3.10",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
  "License :: OSI Approved :: MIT License",
  "Operating System :: OS Independent",
  "Topic :: Scientific/Engineering",
  "Topic :: Scientific/Engineering :: Mathematics",
  "Intended Audience :: Science/Research",
]

[project.urls]
Homepage = "https://github.com/apoorvalal/trex"
Repository = "https://github.com/apoorvalal/trex"

[tool.hatch.build.targets.wheel]
packages = ["trex"]

[project.optional-dependencies]
dev = [
  "pytest>=7.0",
  "pytest-cov",
  "black",
  "isort",
  "mypy",
  "pre-commit",
]
docs = [
  "pdoc>=15.0.0",
  "pandas>=2.0.0",
  "scipy>=1.10.0",
  "sphinx",
  "sphinx-rtd-theme",
  "nbsphinx",
  "jupyter",
  "matplotlib>=3.10.8",
]
test = [
  "pytest>=7.0",
  "pytest-cov",
  "scikit-learn",  # For comparison tests
]

[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = "test_*.py"
python_classes = "Test*"
python_functions = "test_*"
addopts = [
    "-v",
    "--strict-markers",
    "--disable-warnings",
    "--tb=short",
    "--durations=10",
]
markers = [
    "slow: marks tests as slow (deselect with '-m \"not slow\"')",
    "gpu: marks tests that require GPU",
    "integration: marks tests as integration tests",
]

[tool.black]
line-length = 88
target-version = ['py310']

[tool.isort]
profile = "black"
line_length = 88

[tool.mypy]
python_version = "3.10"
warn_return_any = true
warn_unused_configs = true
disallow_untyped_defs = true

[dependency-groups]
dev = [
    "pytest>=9.0.1",
]