Skip to content

Commit 4ac1e24

Browse files
authored
Update pyproject.toml (flagos-ai#2141)
There is no substantial changes in this PR other than pinning some package versions for better alignment.
1 parent 6260729 commit 4ac1e24

1 file changed

Lines changed: 29 additions & 12 deletions

File tree

pyproject.toml

Lines changed: 29 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,58 +1,75 @@
11
[build-system]
2-
requires = ["setuptools>=64.0", "scikit-build-core>=0.11", "pybind11"]
2+
# cmake locks to '>=3.20,<4.0' to sync with FlagTree
3+
requires = ["setuptools>=64.0", "scikit-build-core>=0.11", "pybind11>=3.0.2", "cmake>=3.2.0,<4.0", "ninja>=1.13.0"]
4+
35
build-backend = "scikit_build_core.build"
46

57

68
[project]
9+
710
name = "flag_gems"
11+
812
version = "5.0.1.rc.0"
9-
description = "FlagGems is a function library written in Triton."
13+
14+
description = "FlagGems is a function library written in Triton language."
15+
1016
readme = { file = "README.md", content-type = "text/markdown" }
17+
1118
requires-python = ">=3.8.0"
19+
1220
license = { text = "Apache Software License" }
21+
1322
authors = [
1423
{ name = "Zhixin Li", email = "strongspoon@outlook.com" },
1524
{ name = "Tongxin Bai", email = "waffle.bai@gmail.com" },
1625
{ name = "Yuming Huang", email = "jokmingwong@gmail.com" },
1726
{ name = "Feiyu Chen", email = "iclementine@outlook.com" },
1827
]
28+
1929
classifiers = [
2030
"Development Status :: 3 - Alpha",
2131
"Programming Language :: Python :: 3",
2232
"License :: OSI Approved :: Apache Software License",
2333
]
34+
2435
dependencies = [
36+
# TODO: This is a very old version, should be bumped
2537
"torch>=2.2.0",
26-
"packaging",
27-
"pybind11",
28-
"PyYAML",
29-
"packaging",
30-
"sqlalchemy",
38+
"ninja>=1.13.0",
39+
"packaging>=26.0",
40+
"pybind11>=2.13.1",
41+
"PyYAML>=6.0.3",
42+
"sqlalchemy>=2.0.48",
3143
]
3244

3345
[project.optional-dependencies]
46+
3447
test = [
35-
"pytest>=7.1.0",
48+
"pytest>=9.0.2",
3649
"numpy>=1.26",
37-
"scipy>=1.14",
50+
"scipy>=1.17.0",
3851
]
52+
3953
example = [
4054
"transformers>=4.40.2",
4155
]
4256

4357
[project.urls]
44-
Homepage = "https://github.com/FlagOpen/FlagGems"
4558

46-
[tool.setuptools]
47-
package-dir = { "" = "src" }
59+
Repository = "https://github.com/flagos-ai/FlagGems"
60+
Documentation = "https://flagos-ai.github.io/FlagGems"
61+
"Bug Tracker" = "https://github.com/flagos-ai/FlagGems/issues"
4862

4963
[tool.setuptools.packages.find]
64+
5065
where = ["src"]
5166

5267
[tool.setuptools.package-data]
68+
5369
"flag_gems.runtime" = ["*/**/*.yaml"]
5470

5571
[tool.scikit-build]
72+
5673
build.verbose = true
5774
logging.level = "INFO"
5875
wheel.install-dir = "flag_gems"

0 commit comments

Comments
 (0)