diff --git a/pina/condition/data_condition.py b/pina/condition/data_condition.py index e948305fe..5f5e7d36b 100644 --- a/pina/condition/data_condition.py +++ b/pina/condition/data_condition.py @@ -76,8 +76,9 @@ def __new__(cls, input, conditional_variables=None): # If the input is not of the correct type raise an error raise ValueError( - "Invalid input types. " - "Please provide either torch_geometric.data.Data or Graph objects." + "Invalid input type. Expected one of the following: " + "torch.Tensor, LabelTensor, Graph, Data or " + "an iterable of the previous types." ) def __init__(self, input, conditional_variables=None):