Lr_find error on gan_learner

Hello, I have been building a conditional GAN for a project. But when it comes to running lr_find() or fit() on the learner I get the following error: AttributeError: ‘tuple’ object has no attribute ‘items’. I am running in Google colab with fastai 1.0.61
Following is the error stack I am getting.

AttributeError Traceback (most recent call last)
in ()
----> 1 cgan.lr_find()

5 frames
/usr/local/lib/python3.6/dist-packages/fastai/train.py in lr_find(learn, start_lr, end_lr, num_it, stop_div, wd)
39 cb = LRFinder(learn, start_lr, end_lr, num_it, stop_div)
40 epochs = int(np.ceil(num_it/len(learn.data.train_dl))) * (num_distrib() or 1)
—> 41 learn.fit(epochs, start_lr, callbacks=[cb], wd=wd)
42
43 def to_fp16(learn:Learner, loss_scale:float=None, max_noskip:int=1000, dynamic:bool=True, clip:float=None,

/usr/local/lib/python3.6/dist-packages/fastai/basic_train.py in fit(self, epochs, lr, wd, callbacks)
198 else: self.opt.lr,self.opt.wd = lr,wd
199 callbacks = [cb(self) for cb in self.callback_fns + listify(defaults.extra_callback_fns)] + listify(callbacks)
–> 200 fit(epochs, self, metrics=self.metrics, callbacks=self.callbacks+callbacks)
201
202 def create_opt(self, lr:Floats, wd:Floats=0.)->None:

/usr/local/lib/python3.6/dist-packages/fastai/basic_train.py in fit(epochs, learn, callbacks, metrics)
98 cb_handler.on_epoch_begin()
99 for xb,yb in progress_bar(learn.data.train_dl, parent=pbar):
–> 100 xb, yb = cb_handler.on_batch_begin(xb, yb)
101 loss = loss_batch(learn.model, xb, yb, learn.loss_func, learn.opt, cb_handler)
102 if cb_handler.on_batch_end(loss): break

/usr/local/lib/python3.6/dist-packages/fastai/callback.py in on_batch_begin(self, xb, yb, train)
277 self.state_dict.update(dict(last_input=xb, last_target=yb, train=train,
278 stop_epoch=False, skip_step=False, skip_zero=False, skip_bwd=False))
–> 279 self(‘batch_begin’, call_mets = not self.state_dict[‘train’])
280 return self.state_dict[‘last_input’], self.state_dict[‘last_target’]
281

/usr/local/lib/python3.6/dist-packages/fastai/callback.py in call(self, cb_name, call_mets, **kwargs)
249 if call_mets:
250 for met in self.metrics: self._call_and_update(met, cb_name, **kwargs)
–> 251 for cb in self.callbacks: self._call_and_update(cb, cb_name, **kwargs)
252
253 def set_dl(self, dl:DataLoader):

/usr/local/lib/python3.6/dist-packages/fastai/callback.py in call_and_update(self, cb, cb_name, **kwargs)
240 “Call cb_name on cb and update the inner state.”
241 new = ifnone(getattr(cb, f’on
{cb_name}’)(**self.state_dict, **kwargs), dict())
–> 242 for k,v in new.items():
243 if k not in self.state_dict:
244 raise Exception(f"{k} isn’t a valid key in the state of the callbacks.")

AttributeError: ‘tuple’ object has no attribute ‘items’