import click


def shards_option(func):
    """Option for configuring the shard count."""
    return click.option(
        '--shards', type=int,
        help='Number of shards being used for sharding.')(func)