Skip to content

Commit 3e7c7e2

Browse files
author
Amitabha Roy
committed
Fix long lines.
1 parent 5d96a57 commit 3e7c7e2

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

torch_geometric/data/lightning_datamodule.py

+9-6
Original file line numberDiff line numberDiff line change
@@ -383,15 +383,18 @@ class LightningLinkData(LightningDataModule):
383383
Args:
384384
data (Data or HeteroData): The :class:`~torch_geometric.data.Data` or
385385
:class:`~torch_geometric.data.HeteroData` graph object.
386-
input_train_edges (Tensor or EdgeType or Tuple[EdgeType, Tensor], optional):
386+
input_train_edges (Tensor or EdgeType or Tuple[EdgeType, Tensor]):
387387
The training edges. (default: :obj:`None`)
388-
input_train_edge_label (Tensor, optional): The labels of train edge indices.
389-
input_val_edges (Tensor or EdgeType or Tuple[EdgeType, Tensor], optional):
388+
input_train_edge_label (Tensor, optional):
389+
The labels of train edge indices.
390+
input_val_edges (Tensor or EdgeType or Tuple[EdgeType, Tensor]):
390391
The validation edges. (default: :obj:`None`)
391-
input_val_edge_label (Tensor, optional): The labels of val edge indices.
392-
input_test_edges (Tensor or EdgeType or Tuple[EdgeType, Tensor], optional):
392+
input_val_edge_label (Tensor, optional):
393+
The labels of val edge indices.
394+
input_test_edges (Tensor or EdgeType or Tuple[EdgeType, Tensor]):
393395
The test edges. (default: :obj:`None`)
394-
input_test_edge_label (Tensor, optional): The labels of train edge indices.
396+
input_test_edge_label (Tensor, optional):
397+
The labels of train edge indices.
395398
loader (str): The scalability technique to use (:obj:`"full"`,
396399
:obj:`"link_neighbor"`). (default: :obj:`"link_neighbor"`)
397400
batch_size (int, optional): How many samples per batch to load.

0 commit comments

Comments
 (0)