File size: 2,389 Bytes
1c8d125 87d1677 1c8d125 87d1677 d4f77e3 99d8a1e 1c8d125 9aeb03e 1c8d125 99d8a1e 96e1a32 99d8a1e |
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 |
[project]
name = "TempoPFN"
version = "0.1.0"
description = "Univariate Time Series Forecasting Using Linear RNNs"
authors = [
{ name = "Vladyslav Moroshan" },
{ name = "Julien Siems" },
]
readme = "README.md"
license = { file = "LICENSE" }
requires-python = ">=3.10,<3.13"
dependencies = [
"torch>=2.5.0",
"torchmetrics",
"triton==3.2.0",
"numpy",
"gpytorch",
"flash-linear-attention @ git+https://github.com/fla-org/flash-linear-attention@main",
"scikit-learn",
"pandas",
"gluonts",
"matplotlib",
"notebook",
"datasets",
"ujson",
"huggingface_hub",
]
classifiers = [
"Intended Audience :: Science/Research",
"Intended Audience :: Developers",
"License :: OSI Approved :: Apache Software License",
"Programming Language :: Python",
"Topic :: Software Development",
"Topic :: Scientific/Engineering",
"Operating System :: POSIX",
"Operating System :: Unix",
"Operating System :: MacOS",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
]
[project.optional-dependencies]
dev = [
"wandb",
"build",
"pre-commit",
"pytest>=8",
"ruff",
"mypy",
"commitizen",
"black",
"cupy-cuda12x",
"statsmodels",
"pyo", # Requires portaudio
]
[build-system]
requires = ["setuptools>=68.2.2", "wheel>=0.41.2"]
build-backend = "setuptools.build_meta"
package-dir = {"" = "src"}
[tool.ruff]
line-length = 120
# Set the minimum Python version to target.
target-version = "py312"
# Define the source directories. This matches your project structure.
src = ["src"]
[tool.ruff.lint]
# Select the rules to enable. This is a great starting set.
# E = pycodestyle errors
# F = Pyflakes (e.g., unused imports, undefined names)
# I = isort (import sorting)
# UP = pyupgrade (modernize Python syntax)
# B = flake8-bugbear (common bugs and bad practices)
# C4 = flake8-comprehensions (more efficient comprehensions)
select = ["E", "F", "I", "UP", "B", "C4"]
# You can ignore specific rules here. For example, if you
# don't want to enforce docstrings, uncomment the line below:
# ignore = ["D100", "D101", "D102", "D103"]
[tool.ruff.format]
# Use "black-compatible" formatting.
quote-style = "double"
indent-style = "space"
skip-magic-trailing-comma = false
line-ending = "auto" |