diff --git a/pina/condition/input_target_condition.py b/pina/condition/input_target_condition.py index c67814b..e3924f0 100644 --- a/pina/condition/input_target_condition.py +++ b/pina/condition/input_target_condition.py @@ -72,8 +72,8 @@ class InputTargetCondition(ConditionInterface): raise ValueError( "Invalid input/target types. " - "Please provide either torch_geometric.data.Data, Graph, LabelTensor " - "or torch.Tensor objects." + "Please provide either torch_geometric.data.Data, Graph, " + "LabelTensor or torch.Tensor objects." ) def __init__(self, input, target):