Hi After successfully running the training and the following stage, the script fails in `viz_scrna.py`, line 460 with > AttributeError: module 'matplotlib.cm' has no attribute 'get_cmap'. Probably a `matplotlib` version issue (mine is 3.9.0).