**Is your feature request related to a problem? Please describe.** Related to #593 **Describe the solution you would like** See [comment](https://github.com/graphframes/graphframes/pull/594#issuecomment-3085472410) **Component** <!-- If you do not know what components is it leave empty --> - [x] Scala Core Internal - [ ] Scala API - [ ] Spark Connect Plugin - [ ] Infrastructure - [ ] PySpark Classic - [ ] PySpark Connect **Additional context** . **Are you planning on creating a PR?** <!-- Check the box below with an `x` if you are planning to create a PR for this. --> - [x] I'm willing to make a pull-request