Make CPU and GPU installs separate and remove dev setup

Pbatch · Aug 2, 2024 · d1b93b2 · d1b93b2
1 parent bd4895c
commit d1b93b2
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -14,7 +14,6 @@ requires-python = ">= 3.9"
 dependencies = [
     "flatbuffers>=2.0",
     "numpy>=1.23.5,<2",
-    "onnxruntime-gpu>=1.18.0",
     "Pillow>=10.1.0",
     "protobuf>=3.20.3",
     "scipy>=1.13.1",
@@ -25,16 +24,19 @@ dependencies = [
     "requests>=2.25.1",
     "av",
     "tqdm",
-]
-
-[project.optional-dependencies]
-dev = [
+    "opencv-python==4.10.0.84",
     "pre-commit==3.5.0",
     "black==24.4.0",
     "flake8==7.0.0",
     "isort==5.13.2",
     "pylint==3.1.0",
-    "opencv-python==4.10.0.84",
+]
+[project.optional-dependencies]
+cpu = [
+    "onnxruntime>=1.18.0"
+]
+gpu = [
+    "onnxruntime-gpu>=1.18.0",
 ]
 
 [tool.black]