[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
pyg-team · Dec 15, 2024 · a5c5df7 · a5c5df7
1 parent da1a484
commit a5c5df7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/torch_geometric/explain/config.py b/torch_geometric/explain/config.py
@@ -165,13 +165,13 @@ def __init__(self, mode: Union[ModelMode,
 
         self.return_type = ModelReturnType(return_type)
 
-        if (prediction_threshold is None 
-            and return_type == ModelReturnType.probs):
+        if (prediction_threshold is None
+                and return_type == ModelReturnType.probs):
             prediction_threshold = 0.5
-        elif (prediction_threshold is None 
+        elif (prediction_threshold is None
               and return_type == ModelReturnType.raw):
             prediction_threshold = 0.0
-        elif (prediction_threshold is None 
+        elif (prediction_threshold is None
               and return_type == ModelReturnType.log_probs):
             prediction_threshold = -1.0
         else: