Sample_prior_predictive() failing based solely on `samples` parameter

Looks like pm.sample_prior_predictive() only allows int values passed in for the samples parameter. Trying to pass in (1, n) I get

    draw = pm.sample_prior_predictive(samples=num_trials)
  File "/anaconda3/lib/python3.7/site-packages/pymc3/sampling.py", line 1325, in sample_prior_predictive
    values = draw_values([model[name] for name in names], size=samples)
  File "/anaconda3/lib/python3.7/site-packages/pymc3/distributions/distribution.py", line 369, in draw_values
    size=size)
  File "/anaconda3/lib/python3.7/site-packages/pymc3/distributions/distribution.py", line 463, in _draw_value
    return param.random(point=point, size=size)
  File "/anaconda3/lib/python3.7/site-packages/pymc3/model.py", line 43, in __call__
    return getattr(self.obj, self.method_name)(*args, **kwargs)
  File "/anaconda3/lib/python3.7/site-packages/pymc3/distributions/continuous.py", line 457, in random
    point=point, size=size)
  File "/anaconda3/lib/python3.7/site-packages/pymc3/distributions/distribution.py", line 400, in draw_values
    size=size)
  File "/anaconda3/lib/python3.7/site-packages/pymc3/distributions/distribution.py", line 508, in _draw_value
    output = np.array([func(*v) for v in zip(*values)])
  File "/anaconda3/lib/python3.7/site-packages/pymc3/distributions/distribution.py", line 508, in <listcomp>
    output = np.array([func(*v) for v in zip(*values)])
  File "/anaconda3/lib/python3.7/site-packages/theano/compile/function_module.py", line 813, in __call__
    allow_downcast=s.allow_downcast)
  File "/anaconda3/lib/python3.7/site-packages/theano/tensor/type.py", line 178, in filter
    data.shape))
TypeError: Bad input argument to theano function with name "/anaconda3/lib/python3.7/site-packages/pymc3/distributions/distribution.py:431" at index 0 (0-based). Wrong number of dimensions: expected 0, got 1 with shape (100,).
1 Like