There is error in Edward exmple

http://edwardlib.org/api/ed/models/ParamMixture

i run the codes in jupyter notebook
probs = tf.ones(5) / 5.0
params = {‘mu’: tf.zeros(5), ‘sigma’: tf.ones(5)}
x = ParamMixture(probs, params, Normal)
assert x.shape == ()

it gives tips like this.


TypeError Traceback (most recent call last)
in ()
2 probs = tf.ones(5) / 5.0
3 params = {‘mu’: tf.zeros(5),‘sigma’: tf.ones(5)}
----> 4 x = ParamMixture(probs, params, Normal)
5 assert x.shape == ()

~/anaconda3/lib/python3.6/site-packages/edward/models/param_mixture.py in init(self, *args, **kwargs)
262 # Generate random variable class similar to autogenerated ones from TensorFlow.
263 def init(self, *args, **kwargs):
–> 264 RandomVariable.init(self, *args, **kwargs)
265
266

~/anaconda3/lib/python3.6/site-packages/edward/models/random_variable.py in init(self, *args, **kwargs)
110 self._kwargs[‘collections’] = collections
111
–> 112 super(RandomVariable, self).init(*args, **kwargs)
113
114 self._sample_shape = tf.TensorShape(sample_shape)

~/anaconda3/lib/python3.6/site-packages/edward/models/param_mixture.py in init(self, mixing_weights, component_params, component_dist, validate_args, allow_nan_stats, name)
90 sample_shape=sample_shape,
91 collections=[],
—> 92 **component_params)
93
94 if validate_args:

~/anaconda3/lib/python3.6/site-packages/edward/models/random_variables.py in init(self, *args, **kwargs)
19 # to use _candidate’s docstring, must write a new init method
20 def init(self, *args, **kwargs):
—> 21 _RandomVariable.init(self, *args, **kwargs)
22 init.doc = _candidate.init.doc
23 _params = {‘doc’: _candidate.doc,

~/anaconda3/lib/python3.6/site-packages/edward/models/random_variable.py in init(self, *args, **kwargs)
110 self._kwargs[‘collections’] = collections
111
–> 112 super(RandomVariable, self).init(*args, **kwargs)
113
114 self._sample_shape = tf.TensorShape(sample_shape)

TypeError: init() got an unexpected keyword argument ‘mu’