Error in colab

[prp-e]

I run the notebook and in the "rerank" section I'm getting this:

---------------------------------------------------------------------------

TypeError                                 Traceback (most recent call last)

[<ipython-input-11-cf76c2be222f>](https://localhost:8080/#) in <module>()
     84     img.save(f"./reranked_output/example_{index}.png")
     85 
---> 86 rerank_test(5, 50)

1 frames

[<ipython-input-7-eb7fb2d7bc08>](https://localhost:8080/#) in format_image_grid(img_array)
    151   rows = max_decoder_index + 1
    152 
--> 153   w, h = example_image.size
    154   grid = Image.new('RGB', size=(cols*w, rows*h))
    155   grid_w, grid_h = grid.size

TypeError: cannot unpack non-iterable builtin_function_or_method object

[prp-e]

My bad, I didn't actually run any inference at first. closing the issue.