Skip to content

Commit 925eea5

Browse files
committed
add inference dependencies
1 parent 610d9d4 commit 925eea5

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

pyproject.toml

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,16 @@ dev = ["black", "isort", "pre-commit"]
4141
plot = ["matplotlib", "seaborn"]
4242
wandb = ["wandb"]
4343

44+
inference = [
45+
"numpy",
46+
"pandas",
47+
"torch",
48+
"transformers",
49+
"pysmiles==1.1.2",
50+
"rdkit",
51+
"lightning>=2.5",
52+
]
53+
4454
[tool.setuptools]
4555
include-package-data = true
4656
license-files = ["LICEN[CS]E*"]

0 commit comments

Comments
 (0)