Test failure fix

ShiftLeftSecurity · Mar 6, 2024 · 45296fd · 45296fd
1 parent 076148f
commit 45296fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/codepropertygraph/src/main/scala/io/shiftleft/utils/TimeMetric.scala b/codepropertygraph/src/main/scala/io/shiftleft/utils/TimeMetric.scala
@@ -254,7 +254,7 @@ object TimeMetric {
     stage: Option[String] = None
   ): (Double, Double, Double, Option[(Double, Double)]) = {
     stage match {
-      case Some(st) =>
+      case Some(st) if stagePerformance.contains(st) =>
         val numberProcessor     = stagePerformance(st)
         val (avgCpu, avgMemory) = numberProcessor.getAverage
         val (maxCpu, maxMemory) = numberProcessor.getMax