Assume we're importing from another script, accept dataset object.

--HG--
extra : amend_source : 759cd6c54c57bc13e40729a7ac5b63795ec0b0d9