folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Black formatting

Browse Source
This commit is contained in:
FilippoOlivo
2025-03-12 09:30:17 +01:00
parent e7998b629b
commit 2b65f387e8
4 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pina/condition/condition_interface.py
View File

@@ -45,7 +45,7 @@ class ConditionInterface(metaclass=ABCMeta):
objects is consistent.
:param data_list: List of graph type objects.
:type data_list: torch_geometric.data.Data | Graph|
:type data_list: torch_geometric.data.Data | Graph|
list[torch_geometric.data.Data] | list[Graph]
:raises ValueError: Input data must be either torch_geometric.data.Data

8
pina/condition/data_condition.py
View File

@@ -26,8 +26,8 @@ class DataCondition(ConditionInterface):
types of input data.
:param input: Input data for the condition.
:type input: torch.Tensor | LabelTensor | Graph |
torch_geometric.data.Data | list[Graph] |
:type input: torch.Tensor | LabelTensor | Graph |
torch_geometric.data.Data | list[Graph] |
list[torch_geometric.data.Data] | tuple[Graph] |
tuple[torch_geometric.data.Data]
:param conditional_variables: Conditional variables for the condition.
@@ -63,8 +63,8 @@ class DataCondition(ConditionInterface):
variables (if any).
:param input: Input data for the condition.
:type input: torch.Tensor | LabelTensor | Graph |
torch_geometric.data.Data | list[Graph] |
:type input: torch.Tensor | LabelTensor | Graph |
torch_geometric.data.Data | list[Graph] |
list[torch_geometric.data.Data] | tuple[Graph] |
tuple[torch_geometric.data.Data]
:param conditional_variables: Conditional variables for the condition.

8
pina/condition/input_target_condition.py
View File

@@ -85,13 +85,13 @@ class InputTargetCondition(ConditionInterface):
Initialize the InputTargetCondition, storing the input and target data.
:param input: Input data for the condition.
:type input: torch.Tensor | LabelTensor | Graph |
torch_geometric.data.Data | list[Graph] |
:type input: torch.Tensor | LabelTensor | Graph |
torch_geometric.data.Data | list[Graph] |
list[torch_geometric.data.Data] | tuple[Graph] |
tuple[torch_geometric.data.Data]
:param target: Target data for the condition.
:type target: torch.Tensor | LabelTensor | Graph |
torch_geometric.data.Data | list[Graph] |
:type target: torch.Tensor | LabelTensor | Graph |
torch_geometric.data.Data | list[Graph] |
list[torch_geometric.data.Data] | tuple[Graph] |
tuple[torch_geometric.data.Data]

2
pina/label_tensor.py
View File

@@ -668,7 +668,7 @@ class LabelTensor(torch.Tensor):
Computes the summation of a list of :class:`LabelTensor` instances.
:param list[LabelTensor] tensors: A list of tensors to sum. All
:param list[LabelTensor] tensors: A list of tensors to sum. All
tensors must have the same shape and labels.
:return: A new `LabelTensor` containing the element-wise sum of the
input tensors.