Updated draw_samples for consistent API.