From f8bce58790050fc757784017cf317d5dcbc562f4 Mon Sep 17 00:00:00 2001 From: Maxim Kochurov Date: Sat, 14 Jan 2017 13:41:18 +0300 Subject: [PATCH] fix typo in `as_tensor` function --- pymc3/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pymc3/model.py b/pymc3/model.py index bdfaaf0c15..56241971fe 100644 --- a/pymc3/model.py +++ b/pymc3/model.py @@ -772,6 +772,8 @@ def pandas_to_array(data): return data elif isinstance(data, theano.gof.graph.Variable): return data + elif hasattr(data, '__next__'): + return generator(data) else: return np.asarray(data) @@ -793,8 +795,6 @@ def as_tensor(data, name, model, distribution): constant[data.mask.nonzero()], missing_values) dataTensor.missing_values = missing_values return dataTensor - elif hasattr(data, '__next__'): - return generator(data) else: data = tt.as_tensor_variable(data, name=name) data.missing_values = None