def _shard_keys(self, shard_rank: int, num_shards: int, drop_shard_remainder: bool) -> List[bytes]:
    generated_keys = yogadl.shard_keys(keys=self._keys, shard_index=shard_rank, num_shards=num_shards, sequential=False, drop_shard_remainder=drop_shard_remainder)
    return generated_keys
def _shard_keys(self, shard_rank: int, num_shards: int, drop_shard_remainder: bool) -> List[bytes]:
    generated_keys = yogadl.shard_keys(keys=self._keys, shard_index=shard_rank, num_shards=num_shards, sequential=False, drop_shard_remainder=drop_shard_remainder)
    return generated_keys