Linq IEnumerable batching

EnumerableBatch.cs

// Variant of MoreLinq Batch implementation
// https://code.google.com/p/morelinq/source/browse/MoreLinq/Batch.cs

public static IEnumerable<IEnumerable<T>> Batch<T>(IEnumerable<T> source, int size)
{
    T[] bucket = null;
    var count = 0;

    foreach (var item in source)
    {
        if (bucket == null)
        {
            bucket = new T[size];
        }

        bucket[count++] = item;

        if (count != size)
        {
            continue;
        }

        yield return bucket;

        bucket = null;
        count = 0;
    }

    if (bucket != null && count > 0)
    {
        yield return bucket.Take(count);
    }
}