Actually make `dask.distributed` optional

[leifdenby]

@elbdmi noted that having dask.distributed as optional dependency while not handling the situation where it cannot be imported results in an ImportError of course. I need to handle this 😄

[leifdenby]

closing as this was completed with #27