Fixed name clash in plot.py

This commit is contained in:
Mattéo Delabre 2019-12-18 11:21:14 -05:00
parent 376c7c0bc4
commit 54813c4c46
Signed by: matteo
GPG Key ID: AE3FBD02DC583ABB
1 changed files with 1 additions and 3 deletions

View File

@ -1,4 +1,3 @@
import constellation
from constellation import util
import torch
from matplotlib import pyplot
@ -23,9 +22,8 @@ fig = pyplot.figure()
ax = SubplotZero(fig, 111)
fig.add_subplot(ax)
constellation = model.get_constellation()
util.plot_constellation(
ax, constellation,
ax, model.get_constellation(),
model.channel, model.decoder,
grid_step=0.001, noise_samples=0
)