Update doc condition

Author: FilippoOlivo

pina/condition/condition_interface.py

@@ -45,7 +45,8 @@ def _check_graph_list_consistency(data_list):
         objects is consistent.
 
         :param data_list: List of graph type objects.
-        :type data_list: list(torch_geometric.data.Data) | list(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
             or Graph objects.

pina/condition/data_condition.py

@@ -26,8 +26,10 @@ def __new__(cls, input, conditional_variables=None):
         types of input data.
 
         :param input: Input data for the condition.
-        :type input: torch.Tensor | LabelTensor | Graph |
-            torch_geometric.data.Data
+        :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.
         :type conditional_variables: torch.Tensor | LabelTensor
         :return: Subclass of DataCondition.
@@ -61,7 +63,10 @@ def __init__(self, input, conditional_variables=None):
         variables (if any).
 
         :param input: Input data for the condition.
-        :type input: torch.Tensor or Graph or torch_geometric.data.Data
+        :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.
         :type conditional_variables: torch.Tensor or LabelTensor
 

pina/condition/input_equation_condition.py

@@ -26,7 +26,7 @@ def __new__(cls, input, equation):
         the type of input data.
 
         :param input: Input data. It can be a LabelTensor or a Graph object.
-        :type input: LabelTensor | Graph
+        :type input: LabelTensor | Graph |  list[Graph] | tuple[Graph]
         :param EquationInterface equation: Equation object containing the
             equation function.
         :return: Subclass of InputEquationCondition, based on the input type.
@@ -61,7 +61,7 @@ def __init__(self, input, equation):
         Initialize the InputEquationCondition by storing the input and equation.
 
         :param input: Input data for the condition.
-        :type input: torch.Tensor | Graph
+        :type input: LabelTensor | Graph |  list[Graph] | tuple[Graph]
         :param EquationInterface equation: Equation object containing the
             equation function.
 

pina/condition/input_target_condition.py

@@ -25,11 +25,15 @@ def __new__(cls, input, target):
         the types of input and target data.
 
         :param input: Input data for the condition.
-        :type input: torch.Tensor | Graph | torch_geometric.data.Data | list | \
-            tuple
+        :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 | Graph | torch_geometric.data.Data | list \
-            | tuple
+        :type target: torch.Tensor | LabelTensor | Graph | 
+            torch_geometric.data.Data | list[Graph] | 
+            list[torch_geometric.data.Data] | tuple[Graph] |
+            tuple[torch_geometric.data.Data]
         :return: Subclass of InputTargetCondition
         :rtype: TensorInputTensorTargetCondition | \
             TensorInputGraphTargetCondition | \
@@ -81,9 +85,15 @@ def __init__(self, input, target):
         Initialize the InputTargetCondition, storing the input and target data.
 
         :param input: Input data for the condition.
-        :type input: torch.Tensor | Graph | torch_geometric.data.Data
+        :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 | Graph | torch_geometric.data.Data
+        :type target: torch.Tensor | LabelTensor | Graph | 
+            torch_geometric.data.Data | list[Graph] | 
+            list[torch_geometric.data.Data] | tuple[Graph] |
+            tuple[torch_geometric.data.Data]
 
         .. note::
             If either ``input`` or ``target`` are composed by a list of