Merge branch 'add_package_extras_for_tn_cpu' into 'master'

add package extras See merge request deep-learning/tensornet!17
Qihoo360 · Sep 18, 2024 · d16ea64 · d16ea64
2 parents 9742fb8 + 9a5c14b
commit d16ea64
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,9 +15,10 @@ classifiers=[
   'License :: OSI Approved :: Apache Software License',
   'Programming Language :: Python :: 3.7'
 ]
-dependencies = [
-  "tensorflow-cpu >=2.2,<2.3"
-]
+
+[project.optional-dependencies]
+all = [ "tensorflow >=2.2,<2.3" ]
+cpu = [ "tensorflow-cpu >=2.2,<2.3" ]
 
 [build-system]
 requires = ["setuptools"]