style

rapidsai · Jan 19, 2024 · 0a9c4b8 · 0a9c4b8
1 parent 924e55b
commit 0a9c4b8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/cugraph-equivariant/cugraph_equivariant/nn/tensor_product_conv.py b/python/cugraph-equivariant/cugraph_equivariant/nn/tensor_product_conv.py
@@ -92,8 +92,9 @@ class FullyConnectedTensorProductConv(nn.Module):
     >>>     mlp_channels=None).cuda()
     >>> out = conv2(src_features, edge_sh, edge_emb, graph)
     >>>
-    >>> # Case 3: Same as case 1 but with `mlp_fast_first_layer=True`. The scalar features
-    >>> # from edges, sources and destinations have to be passed in separately.
+    >>> # Case 3: Same as case 1 but with `mlp_fast_first_layer=True`. The scalar
+    >>> #         features from edges, sources and destinations have to be passed
+    >>> #         in separately.
     >>>
     >>> conv3 = FullyConnectedTensorProductConv(in_irreps, sh_irreps, out_irreps,
     >>>     mlp_channels=[6, 16, 16], mlp_fast_first_layer=True).cuda()