From 1f2a193bdc919ad2ee40fab6550db6823d992bd2 Mon Sep 17 00:00:00 2001 From: Andrei Ivanov Date: Mon, 13 Apr 2026 13:59:13 -0700 Subject: [PATCH 1/3] Fix MovieLens dataset incompatibility with SentenceTransformers =5.x --- torch_geometric/datasets/movie_lens.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/torch_geometric/datasets/movie_lens.py b/torch_geometric/datasets/movie_lens.py index 1c9f435c6a74..4e66513bb7f1 100644 --- a/torch_geometric/datasets/movie_lens.py +++ b/torch_geometric/datasets/movie_lens.py @@ -80,8 +80,11 @@ def process(self) -> None: model = SentenceTransformer(self.model_name) with torch.no_grad(): - emb = model.encode(df['title'].values, show_progress_bar=True, - convert_to_tensor=True).cpu() + emb = model.encode( + df['title'].astype(str).tolist(), + show_progress_bar=True, + convert_to_tensor=True + ).cpu() data['movie'].x = torch.cat([emb, genres], dim=-1) From d7ff162a4e58bf7da671cb65acfd47e0bbdea55c Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 13 Apr 2026 21:07:34 +0000 Subject: [PATCH 2/3] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- torch_geometric/datasets/movie_lens.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/torch_geometric/datasets/movie_lens.py b/torch_geometric/datasets/movie_lens.py index 4e66513bb7f1..2d42e6e1244c 100644 --- a/torch_geometric/datasets/movie_lens.py +++ b/torch_geometric/datasets/movie_lens.py @@ -80,11 +80,9 @@ def process(self) -> None: model = SentenceTransformer(self.model_name) with torch.no_grad(): - emb = model.encode( - df['title'].astype(str).tolist(), - show_progress_bar=True, - convert_to_tensor=True - ).cpu() + emb = model.encode(df['title'].astype(str).tolist(), + show_progress_bar=True, + convert_to_tensor=True).cpu() data['movie'].x = torch.cat([emb, genres], dim=-1) From c073bc4ba479c46aedb817fd35508327f0f9e6bd Mon Sep 17 00:00:00 2001 From: Andrei Ivanov Date: Mon, 13 Apr 2026 19:42:23 -0700 Subject: [PATCH 3/3] Updating CHANGELOG.md --- CHANGELOG.md | 1 + 1 file changed, 1 insertion(+) diff --git a/CHANGELOG.md b/CHANGELOG.md index f4125aab5b53..ba142d28c6bd 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -19,6 +19,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/). ### Fixed +- Fix MovieLens dataset incompatibility with SentenceTransformers =5.x ([#10668](https://github.com/pyg-team/pytorch_geometric/pull/10668) - Removed an unnecessary device synchronization in `torch_geometric.utils.softmax` ([#10499](https://github.com/pyg-team/pytorch_geometric/pull/10499)) - Fixed loading of legacy HuggingFace BERT checkpoints ([#10631](https://github.com/pyg-team/pytorch_geometric/pull/10631)) - Fixed `return_attention_weights: bool` being not respected in `GATConv` and `GATv2Conv` ([#10596](https://github.com/pyg-team/pytorch_geometric/pull/10596))