Image classification tutorial bug

Hi, I am trying to follow the tutorial here:

but the following line gives me a bug: Resize(feature_extractor.size)

“Size should be int or sequence. Got <class ‘dict’>”

This is because feature_extractor.size is {‘height’: 224, ‘width’: 224}. Should this be fixed in the tutorial notebook? I see a similar issue for a different notebook that it says was fixed, but it doesn’t seem to have been applied to this one.