Skip to content

Commit efbe81f

Browse files
committed
Update pyproject.toml
Remove TensorFlow and add PyTorch.
1 parent 18922d4 commit efbe81f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ authors = [
99
{ name = "Niels Hulstaert" },
1010
{ name = "Arthur Declercq" },
1111
{ name = "Ralf Gabriels" },
12+
{ name = "Alireza Nameni" },
1213
{ name = "Lennart Martens" },
1314
{ name = "Sven Degroeve" },
1415
]
@@ -31,11 +32,10 @@ keywords = [
3132
]
3233

3334
dependencies = [
34-
"tensorflow>=2.15.0,<3",
35+
"torch>=2.6.0,<3",
3536
"numpy>=1.17",
3637
"pandas>=0.25",
3738
"scikit-learn>=1.2.0",
38-
"deeplcretrainer>=1,<2",
3939
"psm_utils>=0.2.3"
4040
]
4141

0 commit comments

Comments
 (0)