Commit eadf9f31 authored by nd-02110114's avatar nd-02110114
Browse files

🐛 fix error msg

parent 1d54c5fb
Loading
Loading
Loading
Loading
+3 −2
Original line number Diff line number Diff line
@@ -76,9 +76,10 @@ class GraphData:

    if node_pos_features is not None:
      if isinstance(node_pos_features, np.ndarray) is False:
        raise ValueError('pos must be np.ndarray or None.')
        raise ValueError('node_pos_features must be np.ndarray or None.')
      elif node_pos_features.shape[0] != node_features.shape[0]:
        raise ValueError('The length of pos must be the same as the \
        raise ValueError(
            'The length of node_pos_features must be the same as the \
                          length of node_features.')

    self.node_features = node_features