granite-tsfm/pyproject.toml at main · ibm-granite/granite-tsfm · 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
102
[build-system]
requires = ["setuptools>=78.1.1", "wheel", "Cython", "setuptools-scm>=8.0"]

[project]
name = "granite-tsfm"
requires-python = ">=3.11,<3.14"
# version = "0.0.1"
dynamic = ["version"]
authors = [{ name = "IBM" }]
description = "Public notebooks and utilities for TSFM"
readme = "README.md"
license = { file = "LICENSE" }
classifiers = [
    "Development Status :: 4 - Beta",
    "License :: OSI Approved :: Apache Software License",
]
dependencies = [
    "transformers[torch]>=4.57.6,<6",
    "datasets",
    "deprecated",
    "urllib3>=1.26.19", # see https://github.com/urllib3/urllib3/security/advisories/GHSA-34jh-p97f-mpxf
    "numpy<3",
    "torch>=2.10,<2.12",
    "scikit-learn<1.8",
    "pandas>=2.3.3,<4",
    "filelock>=3.20.3",
    "einops>=0.7",
]

# test commit
[tool.setuptools]
packages = ["tsfm_public", "tsfmhfdemos"]

[project.optional-dependencies]


all = ["granite-tsfm[notebooks,testing,dev]"]

notebooks = [
    "jupyter",
    "matplotlib",
    "datasets",
    "ipywidgets",
    "plotly",
    "kaleido",
    "tensorboard",
]
testing = [
    "pytest",
    "pytest-coverage",
    "granite-tsfm[notebooks]",
    "parameterized",
    "pydantic>1,<3",
]
dev = ["granite-tsfm[testing]", "ruff==0.5.7"]

# ogv deployments will already have jupyter
# and we don't want to mess with RH's tested version
ogv = ["datasets", "plotly", "kaleido"]
demos = [
    "granite-tsfm[notebooks]",
    "streamlit",
    "altair<5",
    "plotly",
    "streamlit-aggrid",
    "kaleido",
    "toml",
]
# ###################################################


[tool.setuptools_scm]
version_file = "tsfm_public/_version.py"

[tool.ruff]

line-length = 119
lint.ignore = ["C901", "E501", "E741", "F402", "F823"]
lint.select = ["C", "E", "F", "I", "W"]
extend-exclude = ["tsfm_public/_version.py"]
extend-include = ["*.ipynb"]

# Ignore import violations in all `__init__.py` files.
[tool.ruff.lint.per-file-ignores]
"__init__.py" = ["E402", "F401", "F403", "F811"]

[tool.ruff.lint.isort]
lines-after-imports = 2
known-first-party = ["tsfm_public"]

[tool.ruff.format]
# Like Black, use double quotes for strings.
quote-style = "double"

# Like Black, indent with spaces, rather than tabs.
indent-style = "space"

# Like Black, respect magic trailing commas.
skip-magic-trailing-comma = false

# Like Black, automatically detect the appropriate line ending.
line-ending = "auto"