Single predictions list index out of range in foundation.py

[NZ369]

After loading a model when conduction single predictions, there is a list index out of range issue. The issue is located in foundation.py file, specifically line 120 (else [self.items[i_] for i_ in i]) which assumes batch input.

Input code:
! [ -e /content ] && pip install -Uqq fastbook
import fastbook
fastbook.setup_book()
from fastai.vision.all import *

learn = load_learner('export.pkl')

learn.predict(img)

Error output:

IndexError Traceback (most recent call last)
in
----> 1 learn.predict(img)

20 frames
/usr/local/lib/python3.9/dist-packages/fastcore/foundation.py in (.0)
118 return (self.items.iloc[list(i)] if hasattr(self.items,'iloc')
119 else self.items.array()[(i,)] if hasattr(self.items,'array')
--> 120 else [self.items[i_] for i_ in i])
121
122 def setitem(self, idx, o):

IndexError: list index out of range

[thewunder]

Same exact error trained via:

dls = ImageDataLoaders.from_folder(path, train='training', valid='testing')
learn = vision_learner(dls, resnet18, pretrained=False,
                loss_func=F.cross_entropy, metrics=accuracy)
learn.fit_one_cycle(1, 0.1)