Errors in train_ch3 in tensorflow version softmax-regression-scratch.ipynb about d2l-en HOT 1 OPEN

Here is another bug in the last cell of softmax-regression-concise.ipynb
d2l.train_ch3(net, train_iter, test_iter, loss, num_epochs, trainer)

Here is the bug detail

AssertionError Traceback (most recent call last)
in <cell line: 2>()
1 num_epochs = 2
----> 2 d2l.train_ch3(net, train_iter, test_iter, loss, num_epochs, trainer)

/usr/local/lib/python3.10/dist-packages/d2l/tensorflow.py in train_ch3(net, train_iter, test_iter, loss, num_epochs, updater)
318 animator.add(epoch + 1, train_metrics + (test_acc,))
319 train_loss, train_acc = train_metrics
--> 320 assert train_loss < 0.5, train_loss
321 assert train_acc <= 1 and train_acc > 0.7, train_acc
322 assert test_acc <= 1 and test_acc > 0.7, test_acc

AssertionError: 0.5710129758199056

ImportError Traceback (most recent call last)
/usr/local/lib/python3.10/dist-packages/IPython/core/formatters.py in call(self, obj)
339 pass
340 else:
--> 341 return printer(obj)
342 # Finally look for special method names
343 method = get_real_method(obj, self.print_method)

12 frames
/usr/local/lib/python3.10/dist-packages/matplotlib/backends/backend_svg.py in
16 import matplotlib as mpl
17 from matplotlib import _api, cbook, font_manager as fm
---> 18 from matplotlib.backend_bases import (
19 _Backend, _check_savefig_extra_args, FigureCanvasBase, FigureManagerBase,
20 RendererBase)

ImportError: cannot import name '_check_savefig_extra_args' from 'matplotlib.backend_bases' (/usr/local/lib/python3.10/dist-packages/matplotlib/backend_bases.py)

from d2l-en.