diff --git a/lecture2notes/end_to_end/figure_detection.py b/lecture2notes/end_to_end/figure_detection.py index 9f1477b..0d0b266 100644 --- a/lecture2notes/end_to_end/figure_detection.py +++ b/lecture2notes/end_to_end/figure_detection.py @@ -395,7 +395,7 @@ def all_in_folder( def add_figures_to_ssa(ssa, figures_path): # If the SSA contains frame numbers - if "frame_number" in ssa[0].keys(): + if ssa and "frame_number" in ssa[0].keys(): mapping = frame_number_filename_mapping(figures_path) for idx, slide in enumerate(ssa):