tools.needle_shard module