Style changes

apache · Apr 24, 2014 · e3f6746 · e3f6746
1 parent 07a4f6b
commit e3f6746
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala b/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala
@@ -104,8 +104,11 @@ class DecisionTree (private val strategy: Strategy) extends Serializable with Lo
       // Check whether all the nodes at the current level at leaves.
       val allLeaf = splitsStatsForLevel.forall(_._2.gain <= 0)
       logDebug("all leaf = " + allLeaf)
-      if (allLeaf) break = true // no more tree construction
-      else level += 1
+      if (allLeaf) {
+        break = true // no more tree construction
+      } else {
+        level += 1
+      }
     }
 
     // Initialize the top or root node of the tree.