fix coin flipping in DP; fix #652

edward/models/dirichlet_process.py

@@ -214,7 +214,7 @@ def _sample_n_body(self, k, bools, locs, probs, draws):
     draws = tf.where(bools_tile, locs_k_tile, draws)
 
     # Flip coins according to stick probabilities.
-    flips = Bernoulli(probs_k).sample(n)
+    flips = Bernoulli(probs=probs_k).sample(n)
     # If coin lands heads, assign sample's corresponding bool to False
     # (this ends its "while loop").
     bools = tf.where(tf.cast(flips, tf.bool), tf.zeros_like(bools), bools)