Pickling issues

Hi, I have found a problem in pickling and unpickling the trees. After following through the original streaming example I tried pickling and unpickling a single tree, but the results are not same.

t = forest[0]
with open("a.pkl", "wb") as f:
    pickle.dump(t.to_dict(), f)

t2 = rrcf.RCTree()
with open("a.pkl", "rb") as f:
    t2.load_dict(pickle.load(f))

len(t.leaves)
# 257

len(t2.leaves)
# 238

I thought the issue is while calling pickle.dump as the tree dict is nested, but the documentation says it’ll raise RecursionError if such an object is encountered. So I think the issue could be with the to_dict or load_dict functions. I used both pickle and dill to test this.

[mdbartos]

Greetings, this should now be addressed in #74

[mdbartos]

I’ve confirmed that this is an issue related to duplicate points. Moreover, it only seems to be a problem with the leaves dict–the tree structure itself appears to be ok.

The problem is on lines 816-817 of rrcf.py. to_dict may need to be rewritten to keep track of all the original indices for a given duplicated point.