Computing model output over dataset

not sure… internal API has changed a lot since then. But I guess you don’t need to change too much of your code to make it compatible with the current version.