diff --git a/sql/catalyst/src/main/scala-2.13/org/apache/spark/sql/catalyst/expressions/ExpressionSet.scala b/sql/catalyst/src/main/scala-2.13/org/apache/spark/sql/catalyst/expressions/ExpressionSet.scala index aa85e0c1bbdf..e38deedec6df 100644 --- a/sql/catalyst/src/main/scala-2.13/org/apache/spark/sql/catalyst/expressions/ExpressionSet.scala +++ b/sql/catalyst/src/main/scala-2.13/org/apache/spark/sql/catalyst/expressions/ExpressionSet.scala @@ -132,6 +132,12 @@ class ExpressionSet protected( newSet } + override def --(that: IterableOnce[Expression]): ExpressionSet = { + val newSet = clone() + that.iterator.foreach(newSet.remove) + newSet + } + def map(f: Expression => Expression): ExpressionSet = { val newSet = new ExpressionSet() this.iterator.foreach(elem => newSet.add(f(elem)))