Including images with add_image(...) does not work for me (Python 3.8.10, PyLatex 1.4.1 and Latexmk 4.67)
Error:
LatexMk: Missing input file: 'tmp/{some_heatmap_0.1}.png' from line.
Adding from current working directory works though.
Possible duplicate of issue #320