fix grid

src/tools/plots.py

@@ -26,18 +26,20 @@ def plot_solve_rate(df, task, rows=30, cols=38):
     n = len(values)
     pad_width = rows * cols - n
     
-    # Use masked array to handle NaN values
-    masked_values = np.ma.array(values)
-    masked_values = np.ma.pad(masked_values, (0, pad_width), 'constant', constant_values=np.ma.masked)
+    # Create a masked array
+    masked_values = np.ma.array(np.full(rows * cols, np.nan), mask=True)
+    masked_values[:n] = values
+    masked_values.mask[:n] = False
     masked_values = masked_values.reshape((rows, cols))
 
-    keys = np.pad(keys, (0, pad_width), 'constant', constant_values='').reshape((rows, cols))
+    keys_padded = np.pad(keys, (0, pad_width), 'constant', constant_values='')
+    keys_reshaped = keys_padded.reshape((rows, cols))
 
     hover_text = np.empty_like(masked_values, dtype=object)
     for i in range(rows):
         for j in range(cols):
             if not masked_values.mask[i, j]:
-                hover_text[i, j] = f"{keys[i, j]}<br>Solve Rate: {masked_values[i, j]:.2f}"
+                hover_text[i, j] = f"{keys_reshaped[i, j]}<br>Solve Rate: {masked_values[i, j]:.2f}"
             else:
                 hover_text[i, j] = "NaN"