Using num_nodes=1 raises error

point-cloud-radar / bird-cloud-gnn

This package generates graph representations from point cloud data and fits graph neural networks for classification problem

Apache License 2.0

8 stars 2 forks source link

Using num_nodes=1 raises error #120

Closed abelsiqueira closed 1 year ago

abelsiqueira commented 1 year ago

This error:

    local_tree = KDTree(local_xyz)  # slow
                 ^^^^^^^^^^^^^^^^^
  File ...line 360, in __init__
    super().__init__(data, leafsize, compact_nodes, copy_data,
  File "_ckdtree.pyx", line 558, in scipy.spatial._ckdtree.cKDTree.__init__
ValueError: data must be 2 dimensions

We want to allow it, so we have to create a work around.