Skip to content

Commit 434ab9c

Browse files
authored
Revert "Inference dependencies"
1 parent 46ee848 commit 434ab9c

2 files changed

Lines changed: 11 additions & 11 deletions

File tree

README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ or install it directly with
1818
pip install git+https://github.com/ChEB-AI/python-chebai-graph.git
1919
```
2020

21-
The dependencies `torch`, `torch_geometric` and `torch_scatter` cannot be installed automatically.
21+
The dependencies `torch`, `torch_geometric` and `torch-sparse` cannot be installed automatically.
2222

2323
Use the following command:
2424

@@ -27,8 +27,8 @@ pip install torch torch_scatter torch_geometric -f https://data.pyg.org/whl/torc
2727
```
2828

2929
Replace:
30-
- `${TORCH}` with a PyTorch version (e.g., `2.8.0`; for later versions, check first if they are compatible with torch_scatter and [torch_geometric](https://pytorch-geometric.readthedocs.io/en/latest/install/installation.html))
31-
- `${CUDA}` with `cpu`, `cu118`, `cu121` (or other, depending on your system and CUDA version)
30+
- `${TORCH}` with a PyTorch version (e.g., `2.6.0`; for later versions, check first if they are compatible with torch_scatter and torch_geometric)
31+
- `${CUDA}` with e.g. `cpu`, `cu118`, or `cu121` depending on your system and CUDA version
3232

3333
If you already have `torch` installed, make sure that `torch_scatter` and `torch_geometric` are compatible with your
3434
PyTorch version and are installed with the same CUDA version.

pyproject.toml

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -7,21 +7,21 @@ authors = [
77
]
88
dependencies = [
99
"chebai",
10-
# below packages need to manually installed as mentioned in readme
11-
# torch-geometric
12-
# torch_scatter
10+
"descriptastorus"
1311
]
1412

1513
[project.optional-dependencies]
1614
dev = [
1715
"tox",
18-
"descriptastorus",
16+
"pre-commit",
17+
"black"
1918
]
2019

21-
linters = [
22-
"isort",
23-
"pre-commit",
24-
"black",
20+
inference = [
21+
"chebai" # pip install chebai[inference]
22+
# below packages need to manually installed as mentioned in readme
23+
# torch-geometric
24+
# torch_scatter
2525
]
2626

2727
[build-system]

0 commit comments

Comments
 (0)