cugraph
cugraph copied to clipboard
[FEA] Support Casting WholeGraph Embeddings to PyTorch Tensors and NumPy Arrays in cuGraph Feature Store
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.