Make the torch dep optional

openai · Apr 30, 2024 · 4f392e3 · 4f392e3
1 parent 778caa6
commit 4f392e3
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -43,7 +43,6 @@ dependencies = [
     "statsmodels",
     "termcolor",
     "tiktoken",
-    "torch",
     "tqdm",
     "types-PyYAML",
     "types-tqdm",
@@ -54,12 +53,9 @@ dependencies = [
 repository = "https://github.com/openai/evals"
 
 [project.optional-dependencies]
-formatters = [
-    "black",
-    "isort",
-    "autoflake",
-    "ruff"
-]
+formatters = ["black", "isort", "autoflake", "ruff"]
+
+torch = ["torch"]
 
 [project.scripts]
 oaieval = "evals.cli.oaieval:main"