ImportError: cannot import name 'SpiFlash' from 'litex.soc.cores.spi_flash'

[suppamax]

In caravel.py

from litex.soc.cores.spi_flash import SpiFlash

but there is no such class in litex (at least not in the current version). It has been replaced by LiteSPI.