@esc24 Not your fault, but the doc-string has gone stale and makes reference to `dataset` as an argument, which isn't now passed in ... could you please remove it? Thanks.