Add DatasetEmbedding to root package imports #333

Job	Run time
test / Run tests and report coverage	2m 17s
lint / Run linters	1m 6s
	3m 23s

Provide feedback