diff --git a/utility/startup.py b/utility/startup.py index 540e7c1c..2b974543 100644 --- a/utility/startup.py +++ b/utility/startup.py @@ -88,7 +88,7 @@ def calculate_shard_distribution(total_shards, total_clusters): cluster_kwargs = { 'shard_ids': shard_ids, - 'shard_count': TOTAL_SHARDS, + 'shard_count': len(shard_ids), } return cluster_kwargs