fix shard_args logic, closes #1688

Commit

6 years ago

fix shard_args logic, closes #1688 Co-authored-by: Skye Wanderman-Milne <skyewm@google.com>