diff --git a/pyproject.toml b/pyproject.toml index ecd057e07ec4b5b1f13414b67452593b7c0dc7a6..10346f498ad034771fbc57a0dc405a7df54e9b86 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -109,6 +109,9 @@ optional = true [tool.poetry.group.test.dependencies] pytest = "^6.2.5" +[tool.poetry.group.torch] +optional = true + [tool.poetry.group.torch.dependencies] torch = "^1.13.1"