Fixing the context of ops called from `data.batch()`
Merge pull request #175 from avoidwork/edge

Fixing the context of ops called from `data.batch()`