Make Context.apply() work automatically with DAs.

Now `apply()` extracts the `.key` attribute when using `apply()` on a
DistArray instance.