Skip to content

Commit d7bfc9f

Browse files
committed
Fix matplotlib rendering, update setup.py
1 parent 80ce499 commit d7bfc9f

File tree

3 files changed

+23
-30
lines changed

3 files changed

+23
-30
lines changed

lovely_numpy/repr_rgb.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,10 +66,9 @@ def fig_rgb(x :np.ndarray, # Array to display. [[...], C,H,W] o
6666

6767
ax = fig.add_axes([0,0,1,1])
6868
ax.set_axis_off()
69-
ax.set_xlim(0, x.shape[1]+1)
7069
if close: plt.close(fig)
7170

72-
ax.imshow(x, interpolation="none")
71+
ax.imshow(x, interpolation="none", aspect='equal')
7372
if show: plt.show()
7473

7574
return ax.figure

0 commit comments

Comments
 (0)