style

ibm-granite · Aug 12, 2024 · 9d9e5e5 · 9d9e5e5
1 parent 2d0b61b
commit 9d9e5e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tsfm_public/toolkit/visualization.py b/tsfm_public/toolkit/visualization.py
@@ -232,7 +232,7 @@ def plot_predictions(
     plt.style.use("seaborn-v0_8-whitegrid")
 
     # Adjust figure size and subplot spacing
-    fig, axs = plt.subplots(num_plots, 1, figsize=(10, 2*num_plots))
+    fig, axs = plt.subplots(num_plots, 1, figsize=(10, 2 * num_plots))
     for i, ri in enumerate(random_indices):
         batch = dset[ri]