Skip to content

Commit d16ea64

Browse files
committed
Merge branch 'add_package_extras_for_tn_cpu' into 'master'
add package extras See merge request deep-learning/tensornet!17
2 parents 9742fb8 + 9a5c14b commit d16ea64

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

pyproject.toml

+4-3
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,10 @@ classifiers=[
1515
'License :: OSI Approved :: Apache Software License',
1616
'Programming Language :: Python :: 3.7'
1717
]
18-
dependencies = [
19-
"tensorflow-cpu >=2.2,<2.3"
20-
]
18+
19+
[project.optional-dependencies]
20+
all = [ "tensorflow >=2.2,<2.3" ]
21+
cpu = [ "tensorflow-cpu >=2.2,<2.3" ]
2122

2223
[build-system]
2324
requires = ["setuptools"]

0 commit comments

Comments
 (0)