From 0ea31a9a0a3021927440e1f05841acce703ff33b Mon Sep 17 00:00:00 2001
From: Maurice-Pascal Sonnemann <mpsonnemann@gmail.com>
Date: Thu, 20 Jun 2024 10:36:12 +0200
Subject: [PATCH] fix: build dependencies for pip and poetry are corrected

See: https://github.com/facebookresearch/detectron2/pull/5296
---
 pyproject.toml | 2 ++
 setup.py       | 1 +
 2 files changed, 3 insertions(+)
 create mode 100644 pyproject.toml

diff --git a/pyproject.toml b/pyproject.toml
new file mode 100644
index 0000000000..55b04d0110
--- /dev/null
+++ b/pyproject.toml
@@ -0,0 +1,2 @@
+[build-system]
+requires = ["setuptools", "torch"]
\ No newline at end of file
diff --git a/setup.py b/setup.py
index b51307a625..0e2b7424fa 100644
--- a/setup.py
+++ b/setup.py
@@ -193,6 +193,7 @@ def get_model_zoo_configs() -> List[str]:
         # NOTE: When adding new dependencies, if it is required at import time (in addition
         # to runtime), it probably needs to appear in docs/requirements.txt, or as a mock
         # in docs/conf.py
+        "torch>=2.3.1"
     ],
     extras_require={
         # optional dependencies, required by some features