From 62b4cb023c900f900be8cce859ec6c2f1433b7e5 Mon Sep 17 00:00:00 2001 From: FilippoOlivo Date: Tue, 11 Mar 2025 16:46:11 +0100 Subject: [PATCH] Black formatting on condition --- pina/condition/condition_interface.py | 4 ++-- pina/condition/data_condition.py | 2 +- pina/condition/input_equation_condition.py | 6 +++--- pina/condition/input_target_condition.py | 2 +- 4 files changed, 7 insertions(+), 7 deletions(-) diff --git a/pina/condition/condition_interface.py b/pina/condition/condition_interface.py index b6f4be7..7ff66ed 100644 --- a/pina/condition/condition_interface.py +++ b/pina/condition/condition_interface.py @@ -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. diff --git a/pina/condition/data_condition.py b/pina/condition/data_condition.py index 89fdf22..2b3eeb0 100644 --- a/pina/condition/data_condition.py +++ b/pina/condition/data_condition.py @@ -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 diff --git a/pina/condition/input_equation_condition.py b/pina/condition/input_equation_condition.py index 81f4271..f41b695 100644 --- a/pina/condition/input_equation_condition.py +++ b/pina/condition/input_equation_condition.py @@ -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`. """ diff --git a/pina/condition/input_target_condition.py b/pina/condition/input_target_condition.py index 24dcd3e..c67814b 100644 --- a/pina/condition/input_target_condition.py +++ b/pina/condition/input_target_condition.py @@ -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) """