fix: build dependencies for pip and poetry are corrected

See: facebookresearch#5296
mostlyuseful · Jun 20, 2024 · 0ea31a9 · 0ea31a9
1 parent e8806d6
commit 0ea31a9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -0,0 +1,2 @@
+[build-system]
+requires = ["setuptools", "torch"]
diff --git 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