fix exception

nemccarthy · May 1, 2015 · 42ba873 · 42ba873
1 parent 9117244
commit 42ba873
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/core/src/main/scala/org/apache/spark/SparkContext.scala b/core/src/main/scala/org/apache/spark/SparkContext.scala
@@ -348,9 +348,9 @@ class SparkContext(config: SparkConf) extends Logging with ExecutorAllocationCli
    * Valid log levels include: ALL, DEBUG, ERROR, FATAL, INFO, OFF, TRACE, WARN
    */
   def setLogLevel(logLevel: String) {
-    val logLevels = Seq("ALL", "DEBUG", "ERROR", "FATAL", "INFO", "OFF", "TRACE", "WARN")
-    if (!logLevels.contains(logLevel)) {
-      throw new InvalidArgumentException(
+    val validLevels = Seq("ALL", "DEBUG", "ERROR", "FATAL", "INFO", "OFF", "TRACE", "WARN")
+    if (!validLevels.contains(logLevel)) {
+      throw new IllegalArgumentException(
     s"Supplied level $logLevel did not match one of: ${validLevels.mkString(',')}")
     }
     Utils.setLogLevel(org.apache.log4j.Level.toLevel(logLevel))