fixed bug in map_data returning
@@ -147,9 +147,6 @@ class Poutine(object):
"""
Default pyro.map_data Poutine behavior
"""
- if msg["ret"] is not None:
- return msg["ret"]
- else:
if batch_size is None:
batch_size = 0
assert batch_size >= 0, "cannot have negative batch sizes"
