cugraph icon indicating copy to clipboard operation
cugraph copied to clipboard

[FEA] Support Casting WholeGraph Embeddings to PyTorch Tensors and NumPy Arrays in cuGraph Feature Store

Open alexbarghi-nv opened this issue 1 year ago • 0 comments

The cuGraph feature store currently throws an exception for the torch and numpy backends when a WholeGraph embedding is passed. This should be changed to allow passing a WG embedding, but casting it to the appropriate backend, possibly with a warning.

alexbarghi-nv avatar Jan 04 '24 18:01 alexbarghi-nv