From d1b93b2f9c61737ab36e9ed01ba75bd42dcf5e37 Mon Sep 17 00:00:00 2001
From: peter <peter_batchelor1@hotmail.com>
Date: Fri, 2 Aug 2024 13:40:16 +0100
Subject: [PATCH] Make CPU and GPU installs separate and remove dev setup

---
 pyproject.toml | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index fd8ed89..5d4d997 100644
--- 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]