Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ git submodule update --init --recursive
Then run tests with:

```bash
uv run pytest
uv run pytest -n $(nproc)
```

### Pybind backend from inverted layout
Expand Down
3 changes: 2 additions & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -23,9 +23,10 @@ dev = [
"pybind11~=2.13",
"pytest",
"pytest-asyncio~=1.0",
"pytest-xdist~=3.0",
"setuptools~=80.9",
"black==26.3.0",
"ruff==0.15.18",
"ruff==0.15.19",
"mypy==1.16.0",
"torch>=2.5.0",
"torch-geometric>=2.5.0",
Expand Down
6 changes: 6 additions & 0 deletions test/test_torch_geometric.py
Original file line number Diff line number Diff line change
Expand Up @@ -336,6 +336,8 @@ def test_to_torch_geometric_heterogeneous_graph(conn_db_readonly: ConnDB) -> Non
torch_geometric_data, pos_to_idx, unconverted_properties, edge_properties = (
res.get_as_torch_geometric()
)
# Filter torch.script.jit deprecation warnings in pytorch > 2.10-ish
ws = [w for w in ws if "deprecated" not in str(w.message)]
Comment thread
adsharma marked this conversation as resolved.

assert len(ws) == 13
warnings_ground_truth = {
Expand Down Expand Up @@ -667,6 +669,8 @@ def test_to_torch_geometric_multi_dimensional_lists(
torch_geometric_data, pos_to_idx, unconverted_properties, _ = (
res.get_as_torch_geometric()
)
# Filter torch.script.jit deprecation warnings in pytorch > 2.10-ish
ws = [w for w in ws if "deprecated" not in str(w.message)]
assert len(ws) == 1
assert (
str(ws[0].message)
Expand Down Expand Up @@ -721,6 +725,8 @@ def test_to_torch_geometric_no_properties_converted(
torch_geometric_data, pos_to_idx, unconverted_properties, _ = (
res.get_as_torch_geometric()
)
# Filter torch.script.jit deprecation warnings in pytorch > 2.10-ish
ws = [w for w in ws if "deprecated" not in str(w.message)]
assert len(ws) == 3
warnings_ground_truth = {
"Property personLongString.name of type STRING is not supported by torch_geometric. The property is marked as unconverted.",
Expand Down
Loading