From bef0974a3df40a2fa1d09a397cf0750db432be29 Mon Sep 17 00:00:00 2001 From: toompG Date: Thu, 6 Nov 2025 12:14:20 +0100 Subject: [PATCH] fixed incorrect typehint --- torch_geometric/explain/algorithm/graphmask_explainer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torch_geometric/explain/algorithm/graphmask_explainer.py b/torch_geometric/explain/algorithm/graphmask_explainer.py index 31d9a8e24644..ecca54d6144b 100644 --- a/torch_geometric/explain/algorithm/graphmask_explainer.py +++ b/torch_geometric/explain/algorithm/graphmask_explainer.py @@ -86,9 +86,9 @@ def __init__( epochs: int = 100, lr: float = 0.01, penalty_scaling: int = 5, - lambda_optimizer_lr: int = 1e-2, - init_lambda: int = 0.55, - allowance: int = 0.03, + lambda_optimizer_lr: float = 1e-2, + init_lambda: float = 0.55, + allowance: float = 0.03, allow_multiple_explanations: bool = False, log: bool = True, **kwargs,