Addressing the TorchScript warning that occurs when using the RGCNCon…

…v class.
pyg-team · Oct 11, 2024 · b4e31fd · b4e31fd
1 parent 5a3573f
commit b4e31fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric/nn/conv/rgcn_conv.py b/torch_geometric/nn/conv/rgcn_conv.py
@@ -120,7 +120,7 @@ def __init__(
             in_channels = (in_channels, in_channels)
         self.in_channels_l = in_channels[0]
 
-        self._use_segment_matmul_heuristic_output: Optional[bool] = None
+        self._use_segment_matmul_heuristic_output: Attribute[bool] = None
 
         if num_bases is not None:
             self.weight = Parameter(