Log fatal if energy stats calculation fail

Author: zhouselena

internal/cmd/perfcomp/energystatistics_test.go

@@ -31,7 +31,6 @@ func createTestVectors(start1 int, stop1 int, step1 int, start2 int, stop2 int,
 }
 
 func TestEnergyStatistics(t *testing.T) {
-
 	t.Run("similar distributions should have small e,t,h values ", func(t *testing.T) {
 		x, y := createTestVectors(1, 100, 1, 1, 105, 1)
 		e, tstat, h, _ := getEnergyStatistics(x, y)
@@ -96,5 +95,4 @@ func TestEnergyStatistics(t *testing.T) {
 		assert.NotEqual(t, nil, err)
 		assert.ErrorContains(t, err, "both inputs must be column vectors")
 	})
-
 }

internal/cmd/perfcomp/main.go

@@ -256,7 +256,12 @@ func getEnergyStatsForOneBenchmark(ctx context.Context, rd RawData, coll *mongo.
 
 		estat, tstat, hscore, err := getEnergyStatistics(stableRegionVec, measValVec)
 		if err != nil {
-			return nil, fmt.Errorf("Could not calculate energy stats for test %q, measurement %q: %v", testname, measName, err)
+			log.Fatalf(
+				"Could not calculate energy stats for test %q, measurement %q: %v",
+				testname,
+				measName,
+				err,
+			)
 		}
 
 		zscore := getZScore(measVal, stableRegion.Mean, stableRegion.Std)
@@ -286,7 +291,11 @@ func getEnergyStatsForAllBenchMarks(ctx context.Context, patchRawData []RawData,
 	for _, rd := range patchRawData {
 		energyStats, err := getEnergyStatsForOneBenchmark(ctx, rd, coll)
 		if err != nil {
-			log.Printf("%v\n", err) // Even if energy calculation for one benchmark fails, we still want to show the ones that passed.
+			log.Fatalf(
+				"Could not get energy stats for %q: %v",
+				rd.Info.TestName,
+				err,
+			)
 		} else {
 			allEnergyStats = append(allEnergyStats, energyStats...)
 		}