[yt-dev] Issue #1171: Yt assumes matplotlib._png takes 5 arguments, but it takes only 3 (yt_analysis/yt)

[Anonymous]

New issue 1171: Yt assumes matplotlib._png takes 5 arguments, but it takes only 3
https://bitbucket.org/yt_analysis/yt/issues/1171/yt-assumes-matplotlib_png-takes-5

Anonymous:

Matplotlib version: '1.5.1' (most updated on conda)
Yt version: '3.2.1' (most updated on conda)

matplotlib._png() does not take 5 arguments anymore, as is assumed in png_writer.py.

This gives the following error when calling functions, such as cam.snapshot('myim.png'):

...
    return write_bitmap(out.swapaxes(0, 1), filename)
  File "C:\Anaconda\lib\site-packages\yt\visualization\image_writer.py", line 175, in write_bitmap
    pw.write_png(bitmap_array, filename)
  File "C:\Anaconda\lib\site-packages\yt\utilities\png_writer.py", line 19, in write_png
    _png.write_png(buffer, width, height, filename, dpi)
TypeError: write_png() takes at most 3 arguments (5 given)


Solution:
OLD FUNCTION:
def write_png(buffer, filename, dpi=100):
    width = buffer.shape[1]
    height = buffer.shape[0]
    _png.write_png(buffer, filename, dpi)

NEW FUNCTION (FOR RECENT MATPLOTLIB):
def write_png(buffer, filename, dpi=100):
    _png.write_png(buffer, filename, dpi)