fix index

mlcommons · Oct 24, 2023 · 5fd528a · 5fd528a
1 parent 4441f32
commit 5fd528a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scoring/scoring.py b/scoring/scoring.py
@@ -130,7 +130,7 @@ def get_index_that_reaches_target(workload_df,
   validation_target_reached = validation_series.apply(
       lambda x: op(x, validation_target))
   print(validation_target_reached)
-  target_reached = pd.Series(validation_target_reached[0])
+  target_reached = pd.Series(validation_target_reached)
   # Remove trials that never reach the target
   target_reached = target_reached[target_reached.apply(np.any)]