Currently, shard targeting only works for extension source stages. This ticket should implement shard targeting behavior for extension transform stages and add the appropriate unit/integration tests.