update

rusty1s · May 27, 2024 · 1f2a0ab · 1f2a0ab
1 parent 6ff2026
commit 1f2a0ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_scatter/composite/logsumexp.py b/torch_scatter/composite/logsumexp.py
@@ -38,7 +38,7 @@ def scatter_logsumexp(
     recentered_score = src - max_per_src_element
     recentered_score.masked_fill_(torch.isnan(recentered_score), float('-inf'))
 
-    orig_out = None
+    orig_out: Optional[torch.Tensor] = None
     if out is not None:
         orig_out = out.clone()
         out = out.sub_(max_value_per_index).exp_()