From 16d9a721da016bbadfac13b0fa6dc490cf850e3b Mon Sep 17 00:00:00 2001 From: Andrei Ivanov Date: Mon, 3 Feb 2025 10:51:06 -0800 Subject: [PATCH] Fixing bug in `fused_scatter_reduce` function. --- pyg_lib/ops/scatter_reduce.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyg_lib/ops/scatter_reduce.py b/pyg_lib/ops/scatter_reduce.py index 53ff1e944..52d700951 100644 --- a/pyg_lib/ops/scatter_reduce.py +++ b/pyg_lib/ops/scatter_reduce.py @@ -120,7 +120,7 @@ def fused_scatter_reduce( out = inputs.new(dim_size, len(reduce_list) * num_feats) # Pre-processing: Take care of correct initialization for each reduction: - for reduce in enumerate(reduce_list): + for reduce in reduce_list: assert reduce in REDUCTIONS if reduce == 'min': fill_value = float('inf')