Black formatting on condition
This commit is contained in:
@@ -41,13 +41,13 @@ class ConditionInterface(metaclass=ABCMeta):
|
||||
@staticmethod
|
||||
def _check_graph_list_consistency(data_list):
|
||||
"""
|
||||
Check if the list of :class:`torch_geometric.data.Data`/:class:`Graph`
|
||||
Check if the list of :class:`torch_geometric.data.Data`/:class:`Graph`
|
||||
objects is consistent.
|
||||
|
||||
:param data_list: List of graph type objects.
|
||||
:type data_list: list(torch_geometric.data.Data) | list(Graph)
|
||||
|
||||
:raises ValueError: Input data must be either torch_geometric.data.Data
|
||||
:raises ValueError: Input data must be either torch_geometric.data.Data
|
||||
or Graph objects.
|
||||
:raises ValueError: All elements in the list must have the same keys.
|
||||
:raises ValueError: Type mismatch in data tensors.
|
||||
|
||||
@@ -26,7 +26,7 @@ class DataCondition(ConditionInterface):
|
||||
types of input data.
|
||||
|
||||
:param input: Input data for the condition.
|
||||
:type input: torch.Tensor | LabelTensor | Graph | \
|
||||
:type input: torch.Tensor | LabelTensor | Graph |
|
||||
torch_geometric.data.Data
|
||||
:param conditional_variables: Conditional variables for the condition.
|
||||
:type conditional_variables: torch.Tensor | LabelTensor
|
||||
|
||||
@@ -33,7 +33,7 @@ class InputEquationCondition(ConditionInterface):
|
||||
:rtype: InputTensorEquationCondition | InputGraphEquationCondition
|
||||
|
||||
:raises ValueError: If input is not of type :class:`torch.Tensor`,
|
||||
:class:`LabelTensor`, :class:`Graph`, or
|
||||
:class:`LabelTensor`, :class:`Graph`, or
|
||||
:class:`torch_geometric.data.Data`.
|
||||
"""
|
||||
|
||||
@@ -67,8 +67,8 @@ class InputEquationCondition(ConditionInterface):
|
||||
|
||||
.. note::
|
||||
If ``input`` is composed by a list of :class:`Graph`/
|
||||
:class:`torch_geometric.data.Data` objects, all elements must have
|
||||
the same structure (keys and data types). Moreover, at least one
|
||||
:class:`torch_geometric.data.Data` objects, all elements must have
|
||||
the same structure (keys and data types). Moreover, at least one
|
||||
attribute must be a :class:`LabelTensor`.
|
||||
"""
|
||||
|
||||
|
||||
@@ -87,7 +87,7 @@ class InputTargetCondition(ConditionInterface):
|
||||
|
||||
.. note::
|
||||
If either ``input`` or ``target`` are composed by a list of
|
||||
:class:`Graph`/:class:`torch_geometric.data.Data` objects, all
|
||||
:class:`Graph`/:class:`torch_geometric.data.Data` objects, all
|
||||
elements must have the same structure (keys and data types)
|
||||
"""
|
||||
|
||||
|
||||
Reference in New Issue
Block a user