override toString method on SparkSession builders (#327)

So that cells'output calling methods on builders look nicer
alexarchambault · Jun 20, 2023 · 73d5c21 · 73d5c21
1 parent 2a60af8
commit 73d5c21
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/...ark/src/main/scala/org/apache/spark/sql/almondinternals/NotebookSparkSessionBuilder.scala b/...ark/src/main/scala/org/apache/spark/sql/almondinternals/NotebookSparkSessionBuilder.scala
@@ -20,6 +20,8 @@ class NotebookSparkSessionBuilder(implicit
   commHandler: CommHandler
 ) extends AmmoniteSparkSessionBuilder {
 
+  override def toString = "NotebookSparkSessionBuilder"
+
   override def printLine(line: String, htmlLine: String = null): Unit =
     if (htmlLine == null)
       publish.html(line + System.lineSeparator())

diff --git a/.../main/scala/org/apache/spark/sql/ammonitesparkinternals/AmmoniteSparkSessionBuilder.scala b/.../main/scala/org/apache/spark/sql/ammonitesparkinternals/AmmoniteSparkSessionBuilder.scala
@@ -158,6 +158,8 @@ class AmmoniteSparkSessionBuilder(implicit
   replApi: ReplAPI
 ) extends SparkSession.Builder {
 
+  override def toString = "AmmoniteSparkSessionBuilder"
+
   import AmmoniteSparkSessionBuilder.normalize
 
   def printLine(line: String, htmlLine: String = null): Unit =