comment out edge_weight

compomics · Jun 21, 2024 · 6cede57 · 6cede57
1 parent b1363e5
commit 6cede57
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/molexpress/layers/gatv2_conv.py b/molexpress/layers/gatv2_conv.py
@@ -111,7 +111,7 @@ def call(self, inputs: types.MolecularGraph) -> types.MolecularGraph:
         edge_src = keras.ops.cast(x["edge_src"], "int32")
         edge_dst = keras.ops.cast(x["edge_dst"], "int32")
         edge_state = x.pop("edge_state", None)
-        edge_weight = x.get("edge_weight")
+        # edge_weight = x.get("edge_weight")
 
         if edge_state is None:
             attention_feature = keras.ops.concatenate([