fix: fix small bug in timing logic (#2099)

microsoft · Oct 18, 2023 · 2abfbda · 2abfbda
1 parent 83e199a
commit 2abfbda
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/logging/SynapseMLLogging.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/logging/SynapseMLLogging.scala
@@ -150,19 +150,19 @@ trait SynapseMLLogging extends Logging {
   }
 
   def logFit[T](f: => T, columns: Int, logCertifiedEvent: Boolean = true): T = {
-    logVerb("fit", f, columns, logCertifiedEvent)
+    logVerb("fit", f, Some(columns), logCertifiedEvent)
   }
 
   def logTransform[T](f: => T, columns: Int, logCertifiedEvent: Boolean = true): T = {
-    logVerb("transform", f, columns, logCertifiedEvent)
+    logVerb("transform", f, Some(columns), logCertifiedEvent)
   }
 
-  def logVerb[T](verb: String, f: => T, columns: Int = -1, logCertifiedEvent: Boolean = false): T = {
+  def logVerb[T](verb: String, f: => T, columns: Option[Int] = None, logCertifiedEvent: Boolean = false): T = {
     val startTime = System.nanoTime()
     try {
-      // Begin emitting certified event.
-      logBase(verb, Some(columns), Some((System.nanoTime() - startTime) / 1e9), logCertifiedEvent)
-      f
+      val ret = f
+      logBase(verb, columns, Some((System.nanoTime() - startTime) / 1e9), logCertifiedEvent)
+      ret
     } catch {
       case e: Exception =>
         logErrorBase(verb, e)