Also, add a ScalaDoc to this and add `:: Experimental ::` at the beginning of the doc and the annotation `@ExperimentalAPI`. Then it will be marked experimental in the docs, similar to https://spark.apache.org/docs/latest/api/scala/index.html#org.apache.spark.FutureAction.