Skip to content

Commit c235810

Browse files
authored
fix msg error datacondition (#729)
1 parent c76cf21 commit c235810

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pina/condition/data_condition.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,8 +76,9 @@ def __new__(cls, input, conditional_variables=None):
7676

7777
# If the input is not of the correct type raise an error
7878
raise ValueError(
79-
"Invalid input types. "
80-
"Please provide either torch_geometric.data.Data or Graph objects."
79+
"Invalid input type. Expected one of the following: "
80+
"torch.Tensor, LabelTensor, Graph, Data or "
81+
"an iterable of the previous types."
8182
)
8283

8384
def __init__(self, input, conditional_variables=None):

0 commit comments

Comments
 (0)