[variational] plot needed kwargs

This commit is contained in:
mzwiessele 2015-04-08 08:24:00 +02:00
parent 66c7da879a
commit 1efa842130

View file

@ -141,7 +141,7 @@ class NormalPosterior(VariationalPosterior):
holds the means and variances for a factorizing multivariate normal distribution holds the means and variances for a factorizing multivariate normal distribution
''' '''
def plot(self, *args): def plot(self, *args, **kwargs):
""" """
Plot latent space X in 1D: Plot latent space X in 1D:
@ -150,8 +150,7 @@ class NormalPosterior(VariationalPosterior):
import sys import sys
assert "matplotlib" in sys.modules, "matplotlib package has not been imported." assert "matplotlib" in sys.modules, "matplotlib package has not been imported."
from ...plotting.matplot_dep import variational_plots from ...plotting.matplot_dep import variational_plots
import matplotlib return variational_plots.plot(self, *args, **kwargs)
return variational_plots.plot(self,*args)
class SpikeAndSlabPosterior(VariationalPosterior): class SpikeAndSlabPosterior(VariationalPosterior):
''' '''