update

minmingzhu · Jul 30, 2024 · 984ac8f · 984ac8f
1 parent d2e8847
commit 984ac8f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/mllib-dal/src/main/scala/com/intel/oap/mllib/stat/CorrelationDALImpl.scala b/mllib-dal/src/main/scala/com/intel/oap/mllib/stat/CorrelationDALImpl.scala
@@ -133,13 +133,12 @@ class CorrelationDALImpl(
       }
       ret
     }.collect()
+    // Make sure there is only one result from rank 0
+    assert(results.length == 1)
     logInfo(s"CorrelationDAL compute end")
     corTimer.record("Training")
     corTimer.print()
 
-    // Make sure there is only one result from rank 0
-    assert(results.length == 1)
-
     val correlationMatrix = results(0)
 
     correlationMatrix